CẶP SỐ

View as PDF

Points: 100 (p) Time limit: 1.0s Memory limit: 512M Input: stdin Output: stdout

Cho một dãy \(N\) số nguyên \(a_1,a_2,..,a_n\).
Yêu cầu: Đếm số lượng cắp \(i,j (1 \leq i \leq j \leq N)\) thỏa mãn \(a_j - a_i = i^2 + j^2\).

INPUT

  • Dòng đầu tiên chứa số nguyên dương \(N (1 \leq N \leq 10^5)\)
  • Dòng thứ \(2\) chứa \(N\) số nguyên dương \(a_1,a_2,...,a_N (|a_i| \leq 10^{12})\).

OUTPUT

  • In ra số lượng cặp thỏa mãn yêu cầu.

Example

Test 1

Input
5
4 9 6 29 30
Output
3

Ràng buộc

  • Subtask \(1\) (\(70\%\) test): Có \((N \leq 10^3)\)
  • Subtask \(2\) (\(30\%\) test): Không có ràng buộc gì thêm.

Comments

There are no comments at the moment.