Source Code
#define _CRT_SECURE_NO_WARNINGS
#include <iostream>
#include <cmath>
#include <iomanip>
#include <algorithm>
#include<functional>
#include <string>
#include <vector>
#include <stack>
#include <queue>
#include <deque>
#include <map>
#include <unordered_map>
#include <set>
#include <unordered_set>
#include<iterator>
#include <stdio.h>
#include <string.h>
#define T int t=0;cin>>t;while(t--)
#define F(i,n) for(int i=0;i<n;i++)
#define fo(i,a)for(auto&i:a)
#define all(a) a.begin(),a.end()
#define mem(x,y) memset(x,y,sizeof(x))
#define ll long long
#define ld long double
#define endl "\n"
#define vl vector<ll>
#define vi vector<int>
#define cost first
#define from second.first
#define to second.second
#define sz(X) X.size()
using namespace std;
const ll mod = 1000000007;
void code();
///*-*-*-*-*-*-*-*-*-*-*-*-*-*-*>> # str main # <<*-*-*-*-*-*-*-*-*-*-*-*-*-*-*///
int main()
{
    code();
    T
    {
        ll n, k;
        cin >> n >> k;
        ll sum1=0,ans=0;
        vl a(n);
        ll sum=0,an=0;
        fo(i,a)
        {
            cin >> i;

            sum+=i;
            if(sum%k==0)ans++,an++;

        }
        for(int i=n-1;i>=0;i--)
        {
            sum1+=a[i];
            if(sum1%k==0)ans+=an;
            if(sum%k==0)an--;
            sum-=a[i];
        }
        cout<<ans<<endl;
    }
}
///*-*-*-*-*-*-*-*-*-*-*-*-*-*-*>> # end main # <<*-*-*-*-*-*-*-*-*-*-*-*-*-*-*///
void code()
{
    std::ios_base::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);
    cout << fixed << setprecision(6);
}
Copy
Number of Ways ABD_ALMABOUD
GNU G++17
18 ms
312 KB
Wrong Answer