ツバサの備忘録

主に備忘録代わりに精進記録を載せていくつもりです。

ABC166 E - This Message Will Self-Destruct in 5s

問題
提出コード

解法

もちろん、ペア(i,j)を全探索することは難しいです。
ペアとしてカウントされる条件を詳しく見ていきます。
i \lt jとしたとき、カウントされる条件は、

  • j - i = A_{i} + A_{j}

となります。
これを式変形すると...

  • j - A_{j} = A_{i} + i

となります。
jより小さくて、jとペアになれるiの個数は、上の条件を満たすiの個数になります。
あとは、jについて前から見ていき、A_{i} + iをそれぞれの値ごとにいくつ存在しているか、を見ればよいです。
やることは下の記事にある問題(と、そのページに貼られている類題)とほとんど同じです。

emtubasa.hateblo.jp