题意:
给出一个长度为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统计答案即可。