có ai giúp mình với ạ,c++ or python gì cũng đc

Cặp (a[i], a[j]) sẽ hợp lệ nếu tổng của chúng chia hết cho 3. Điều này xảy ra khi:
Cả hai số đều chia hết cho 3 (cả hai có phần dư bằng 0).
Một số có phần dư là 1 và số còn lại có phần dư là 2.

#include <bits/stdc++.h>
using namespace std;

long long tt, n , s = 0, a[3] = {};
int main() {
  cin >> tt;
  while (tt--)
  {
    cin >> n;
    a[n % 3]++;
  }
  cout << (a[0] * (a[0] - 1) / 2) + a[1] * a[2];
  return 0;
}

#include <bits/stdc++.h>
using namespace std;

int main() {
    long long n;
    cin >> n; 
    long long count[3] = {0}; 

    for (long long i = 0; i < n; i++) {
        int x;
        cin >> x; 
        count[x % 3]++; 
    }

    long long d = (count[0] * (count[0] - 1)) / 2; 
    d += count[1] * count[2]; 

    cout << d << endl; 
    return 0;
}