k倍区间 - 蓝桥云课 (树状数组)

k倍区间 - 蓝桥云课 (lanqiao.cn)

#include <iostream>
#include <bits/stdc++.h>
#include <cstring>
#include <vector>
#include <queue>
#include <algorithm>
#define x first
#define y second
#define fu(i,a,b) for(int i=a;i<=b;i ++ )
#define fd(i,a,b) for(int i=a;i>=b;i -- )
#define endl '\n'
#define ms(x,y) memset(x,y,sizeof x)
#define ios ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
using namespace std;

typedef long long LL;
typedef vector<vector<LL>> VVL;
typedef vector<LL> VL;
typedef vector<int> VI;
typedef pair<int,int> PII;
typedef pair<double,double> PDD;
typedef pair<double,int> PDI;
typedef pair<char,int> PCI;
typedef pair<string,int> PSI;
typedef pair<int,string> PIS;
typedef pair<LL,LL> PLL;
typedef __int128 i128;
typedef unsigned long long ULL;
const int N = 1e6+ 10,M = N * 4,INF = 0x3f3f3f3f ;
const int mod = 1e9; 
const double eps = 1e-8;

int n,m;
unordered_map<int,vector<int>> tr;

int find(VL &t,LL x)
{
    return lower_bound(t.begin(),t.end(),x) - t.begin() +  1; 
}
int lowbit(int x)
{
    return x  & -x ;
}
void add(VI & t,int x,LL c)
{
    for(int i=x;i<t.size();i+=lowbit(i)) t[i] += c;
}

LL sum(VI &t,int x)
{
    LL res =0 ;
    for(int i=x;i;i-=lowbit(i))    res += t[i];
    return res;
}

inline void solve()
{
    cin >> n >> m ;
    unordered_map<int,vector<LL>> cnt;

    vector<LL> s(n+10);
    LL res =0 ;
    fu(i,1,n)
    {
        cin >> s[i],s[i] += s[i-1];
        int t = (s[i] % m + m ) % m;
        cnt[t].push_back(s[i]);
// 将模m的相同的正余数 的 数 放进 同一组中
    }
    
    for(auto &t: cnt )
    {
        VL &ve =t.y ; // 组内排序去重
        sort(ve.begin(),ve.end());
        ve.erase(unique(ve.begin(),ve.end()),ve.end());
        tr[t.x] = VI(ve.size()+1,0);
// 为这个余数 新开一段哈希表
    }
    
    
    fu(i,1,n)
    {
        int t =(s[i] % m + m ) % m;
        int x = find(cnt[t],s[i]);
      // 找到这个数 在组内的位置  
        if(!t && s[i] >= 0 ) res ++ ;
// 这段[1,i] 刚好是m的倍数，且这段值为正数
        res += sum(tr[t],x);
// 同一组中模m的余数相同，前面比x小的数 i 都可以和x组成一段k的正整数倍区间
// s[i] %m == s[x] % m  s[i] <= s[x]
        add(tr[t],x,1);
        
        
    }
        
    
        cout << res <<endl;
        
}


signed main()
{
//  freopen("1.txt","w",stdout);
     ios

    int t=1;
//    cin>>t;
    int now = 1;
    while(t -- )
    {
//      cout<<"Case "; 
//      cout<<"Scenario #"; 
//      cout<< now ++ <<": ";
//      cout<< now ++ <<": \n";
        solve();
    }


    return 0;
}