fork download
  1. #include <cmath>
  2. #include <iostream>
  3. #include <vector>
  4. #include <algorithm>
  5. #include <string>
  6. #include <set>
  7. #include <map>
  8. #include <list>
  9. #include <time.h>
  10. #include <math.h>
  11. #include <random>
  12. #include <deque>
  13. #include <queue>
  14. #include <cassert>
  15. #include <unordered_map>
  16. #include <unordered_set>
  17. #include <iomanip>
  18. #include <bitset>
  19. #include <sstream>
  20. #include <chrono>
  21. #include <cstring>
  22.  
  23. using namespace std;
  24.  
  25. typedef long long ll;
  26.  
  27. #ifdef iq
  28. mt19937 rnd(228);
  29. #else
  30. mt19937 rnd(chrono::high_resolution_clock::now().time_since_epoch().count());
  31. #endif
  32.  
  33. int main() {
  34. #ifdef iq
  35. freopen("a.in", "r", stdin);
  36. #endif
  37. ios::sync_with_stdio(0);
  38. cin.tie(0);
  39. int t;
  40. cin >> t;
  41. while (t--) {
  42. int n;
  43. cin >> n;
  44. vector <int> a(n);
  45. int orz = 0;
  46. int sorz = 0;
  47. for (int i = 0; i < n; i++) {
  48. cin >> a[i];
  49. if (a[i] == 2) {
  50. orz++;
  51. }
  52. if (a[i] == 0) {
  53. sorz++;
  54. }
  55. }
  56. cout << orz * (ll) (orz - 1) / 2 + sorz * (ll) (sorz - 1) / 2 << '\n';
  57. }
  58. }
Success #stdin #stdout 0s 4504KB
stdin
Standard input is empty
stdout
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260
7260