Description

波特石路最近在看書的時候,看到了費馬最後定理:

「$x^ n+y^ n=z^ n$ 在 $n\ge 3$ 的時候,不存在正整數解 $(x,y,z)$。」

這個定理在 17 世紀時被提出,但直到 1995 年才被安德魯·懷爾斯及其學生理查·泰勒證明完畢。

波特石路認為這個定理敘述中「正」整數是不重要的,於是他提出了以下的猜想:

「$x^ n+y^ n=z^ n$ 在 $n\ge 3$ 的時候,不存在整數解 $(x,y,z)$。」

他告訴他哥哥 -- 波路特石這個新的猜想,但數學明顯強很多的波路特石馬上就否定了波特石路的猜想。

以上是《波波歷險記》的部分內容。

現在開始才是真正的題目:

給定 $m$ 個整數 $a_1,a_2,\ldots,a_m$ 和一個正整數 $n\ge 3$,請問有多少正整數三元組 $(i,j,k)$ 滿足 $1\le i,j,k\le m$ 且 $a_i^ n+a_j^ n=a_k^ n$?

Input Format

第一行輸入兩個正整數 $m,n$。

第二行輸入 $m$ 個整數 $a_1,a_2,\ldots,a_m$。

  • $3\le m\le 2\times 10^ 5$
  • $3\le n\le 10^ 9$
  • $|a_i|\le 10^ 5$

Output Format

輸出滿足條件的 $(i,j,k)$ 個數。

Sample Input 1

4 4
0 -1 1 1

Sample Output 1

19

Sample Input 2

4 5
0 -1 1 1

Sample Output 2

15

Hints

Problem Source

Subtasks

No. Testdata Range Constraints Score
1 0~1 範例測資 0
2 2~14 $a_i\ge 0$ 30
3 0~29 無額外限制 70

Testdata and Limits

No. Time Limit (ms) Memory Limit (VSS, KiB) Output Limit (KiB) Subtasks
0 1000 262144 65536 1 3
1 1000 262144 65536 1 3
2 1000 262144 65536 2 3
3 1000 262144 65536 2 3
4 1000 262144 65536 2 3
5 1000 262144 65536 2 3
6 1000 262144 65536 2 3
7 1000 262144 65536 2 3
8 1000 262144 65536 2 3
9 1000 262144 65536 2 3
10 1000 262144 65536 2 3
11 1000 262144 65536 2 3
12 1000 262144 65536 2 3
13 1000 262144 65536 2 3
14 1000 262144 65536 2 3
15 1000 262144 65536 3
16 1000 262144 65536 3
17 1000 262144 65536 3
18 1000 262144 65536 3
19 1000 262144 65536 3
20 1000 262144 65536 3
21 1000 262144 65536 3
22 1000 262144 65536 3
23 1000 262144 65536 3
24 1000 262144 65536 3
25 1000 262144 65536 3
26 1000 262144 65536 3
27 1000 262144 65536 3
28 1000 262144 65536 3
29 1000 262144 65536 3