Codeforces Round #341 (Div. 2) B. Wet Shark and Bishops

もうすぐ期末試験。

1000x1000 のチェス盤の上に n 個のビショップが置かれている。通常のビショップとは異なり駒を飛び越えることもできる。互いに取られる位置にあるビショップの対はいくつあるか。

ビショップは将棋の角と同じで、斜め 4 方向に遮るものがない限り進める。ただし今回は遮るものがあるかは考えなくても良い。

互いに取られる位置にあるビショップというのは、x+y が等しいか x-y が等しいかのどちらか一方である。ゆえに求める場合の数は次の総和になる。

配列だと負のインデックスの扱いが面倒なので map を使った。

間に駒があっても大丈夫なのは直感に反する。