fork(2) download
  1. #include <bits/stdc++.h>
  2. using namespace std;
  3.  
  4. void compute(){
  5. int a, b, c, d, m, n; cin>>a>>b>>c>>d>>m>>n;
  6. vector<int> v(n);
  7. vector<vector<int> > vv(5);
  8. for(int i=0; i<n; i++){
  9. cin>>v[i];
  10. vv[(v[i]*v[i])%m].push_back(v[i]);
  11. }
  12. long long res = 0;
  13. for(int i=0; i<n; i++){
  14. int y = (a*v[i]*v[i]*v[i] + b*v[i]*v[i] + c*v[i] + d)%m;
  15. if(y>=0)
  16. res += vv[y].size();
  17. }
  18. cout<<res<<"\n";
  19. }
  20.  
  21. int main() {
  22. int t; cin>>t;
  23. while(t--) compute();
  24. return 0;
  25. }
Success #stdin #stdout 0s 15240KB
stdin
1
2 1 -1 3 5
5
10 2 3 14 12
stdout
2