P6393 数学

P6393 隔离的日子

先化简式子，

\[\begin{aligned} (a_j - b_i)(b_i + b_j) &= a_j b_i + a_i b_j \\ a_j b_j + a_j b_i - b_i b_j - {b_i}^2 &= a_j b_i + a_i b_j \\ b_j(-a_i + a_j - b_i) &= {b_i}^2 \end{aligned} \]

可知 \((-a_i + a_j - b_i) \mid {b_i}^2\)，则可以在枚举 \(i\) 时检查 \({b_i}^2\) 的所有因子，加以 map / unordered_map 维护下标即可。

具体参见 代码。