fork download
  1. #include<bits/stdc++.h>
  2. #define REP(i,n) for (int i = 1; i <= n; i++)
  3. #define mod 1000000007
  4. #define pb push_back
  5. #define ff first
  6. #define ss second
  7. #define ii pair<int,int>
  8. #define vi vector<int>
  9. #define vii vector<ii>
  10. #define lli long long int
  11. #define INF 1000000000
  12. #define endl '\n'
  13. const double PI = 3.141592653589793238460;
  14. typedef std::complex<double> Complex;
  15. typedef std::valarray<Complex> CArray;
  16.  
  17. using namespace std;
  18. int main()
  19. {
  20. lli t , ar[3] , n , x , y;
  21. cin>>t;
  22. while(t--)
  23. {
  24. cin>>n , ar[0] = ar[1] = ar[2] = 0;
  25. REP(i , n)
  26. {
  27. cin>>x;
  28. if(x <= 2) ar[x]++;
  29. }
  30.  
  31. x = (ar[0] * (ar[0] - 1)) / 2;
  32. y = (ar[2] * (ar[2] - 1)) / 2;
  33. cout<<x + y<<endl;
  34. }
  35. }
  36.  
Success #stdin #stdout 0s 4288KB
stdin
2
3
2 4 2
3
0 2 3
stdout
1
0