题意:
给出一个长度为n的数组a,两个整数p和k。
询问有多少数对\((x,y)(1 \leq x < y \leq n)\)使得\((a_x^2+a_y^2)(a_x+a_y)\)对p取模的结果为k。
保证p是质数。
题解:
\((a_x^2+a_y^2)(a_x+a_y)\%p=k\)
\((a_x^2+a_y^2)(a_x+a_y)(a_x-a_y)\%p=(a_x-a_y)k\%p\)
\((a_x^2+a_y^2)(a_x^2-a_y^2)\%p=(a_x-a_y)k\%p\)
\((a_x^4-a_y^4)\%p=(a_x-a_y)k\%p\)
\((a_x^4-ka_x)\%p=(a_y^4-ka_y)\%p\)
对每个a_x统计答案即可。