ARC047 B問題:同一円周上

問題. 同一円周上

平面上に  N 個の格子点がある.マンハッタン距離でこれらの格子点からの距離が等しくなる格子点を求めよ.

制約 1 \le N \le 10^5,  -10^9 \le x_i, y_i \le 10^9

解法.1次変換

 i 番目の格子点  (x_i, y_i) (x_i + y_i, x_i - y_i) に1次変換する.この1次変換は各点に対して,原点を中心に時計回りに  \pi / 4 回転して,原点を中心に  \sqrt 2 倍に拡大したあとに, x 軸に関して折り返しをした変換である. (x_i, y_i) の1次変換を行列で表現すると,
$$
\begin{pmatrix}
1 & 0 \\\
0 & -1
\end{pmatrix}
\begin{pmatrix}
\sqrt 2 & 0 \\\
0 & \sqrt 2
\end{pmatrix}
\begin{pmatrix}
\cos \frac{\pi}{4} & \sin \frac{\pi}{4} \\\
-\sin \frac{\pi}{4} & \cos \frac{\pi}{4}
\end{pmatrix}
\begin{pmatrix}
x_i \\\
y_i
\end{pmatrix}
$$
となる.
マンハッタン距離上で格子点  p から等しい距離にある格子点全体は菱形をしているが,上の1次変換を行うと等しい距離にある格子点全体は  p の変換後の点を中心とした正方形をしている.またこの1次変換は格子点を格子点に写す全単射となる.ただし,この1次変換は 等長写像 ではないことに注意が必要である(∵  \sqrt 2 倍に拡大).下の図では,丸から同じ色の正方形への1次変換を表している.丸は中心座標からマンハッタン距離で4となる格子点である.
   f:id:tatanaideyo:20190508225348p:plain

与えられた格子点全体  P に対して1次変換を行った後の格子点全体  P' について考える. P' から等しい距離にある点  p というのは,上で考察したとおり,ある長さ  d に対して  p' \in P' を中心とした1辺の長さが  d の正方形全体の交点となる.1次元で考えると正方形は区間となり, N 個の区間の端点が1点で交差する場合というのは座標値が最小の点と最大の点の区間が交差する初めての場所となる.あとはこの候補となる点を全探索すれば答えが求まる.

計算時間 O(N)

他の人のブログで45°回転と書いていたので一瞬迷った.