Source Code
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef long double ld;

#define forr(_a, _b, _c) for(_a = (_b); _a <= (_c); ++_a)
#define ford(_a, _b, _c) for(_a = (_b) + 1; _a --> (_c);)
#define forf(_a, _b, _c) for(_a = (_b); _a < (_c); ++_a)
#define mp make_pair
#define fi first
#define se second
#define vi vector
#define sz(_v) _v.begin(), _v.end()
#define mask(_x) (1ll << (_x))
#define bit(_x,_y) (((_x) >> (_y)) & 1)

string yes[2] = {"NO\n","YES\n"};
const ld eps = ld(1e-7);
const ll inf = ll(1e16) + 1;
const ll mod = ll(1e9) + 7;

const int N = 2e5 + 5;
int n, i;
ll k, a[N], f[N], g[N], o;
map<ll,ll> dd;

void solve() {
	cin >> n >> k;
	forr(i,1,n) {
		cin >> a[i];
		f[i] = (f[i - 1] + a[i]) % k;
	}
	g[n + 1] = o = 0;
	dd.clear();
	ford(i,n,1) {
		g[i] = (g[i + 1] + a[i]) % k;
		++dd[g[i]];
	}
	forr(i,1,n) {
		--dd[g[i]];
		o += dd[(k - f[i]) % k];
	}
	cout << o << "\n";
}

void precalc() {
}

int main() {
	ios_base::sync_with_stdio(false); cin.tie(NULL);
	#ifdef umf
		freopen("test.inp","r",stdin);
		freopen("test.out","w",stdout);
	#endif

	int tc = 1;
	cin >> tc;
	precalc();
	while(tc --> 0) {
		solve();
	}

	return 0;
}
Copy
Number of Ways thqdothabit
GNU G++17
337 ms
30.8 MB
Accepted