1. 程式人生 > >GYM 100741A Queries

GYM 100741A Queries

樹狀數組 ++ color 樹狀 sca pan 數組 logs 代碼

傳送門

題目大意:

一個長度為n的序列,q次三種操作

+p r:下標為p的數+r

-p r:下標為p的數-r

s l r mod [L,R]中有多少數%m=mod,m已經給出

題解:

開十個樹狀數組

代碼

我的

#include<iostream>
#include<cstdio>
using namespace std;

int n,m,q,l,r,od,p,mod,a[100];
struct Tree{
    int tre[1001];
}s[10];

int lowbit(int x){
    return x&(-x);
}

void add(int
k,int p){ while(p<=n){ s[k].tre[p]++; p+=lowbit(p); } } void modify(int k,int p,int x){ while(p<=n){ s[k].tre[p]+=x; p+=lowbit(x); } } int sum(int l,int r,int mod){ int cntl=0,cntr=0;l--; while(l){ cntl+=s[mod].tre[l]; l-=lowbit(l); }
while(r){ cntr+=s[mod].tre[r]; r-=lowbit(r); } return cntr-cntl; } int main(){ scanf("%d%d%d",&n,&m,&q); for(int i=1;i<=n;i++){ scanf("%d",&a[i]); add(a[i]%m,i); } for(int i=1;i<=q;i++){ scanf("%d",&od);
if(od==1){ scanf("%d%d",&p,&r); modify(a[p]%m,p,-1); a[p]+=r; modify(a[p]%m,p,1); } if(od==2){ scanf("%d%d",&p,&r); modify(a[p]%m,p,-1); a[p]-=r; modify(a[p]%m,p,1); } if(od==3){ scanf("%d%d%d",&l,&r,&mod); printf("%d\n",sum(l,r,mod)); } } return 0; }

丁神的

#include<bits/stdc++.h>
#define N 10010
#define ll long long
using namespace std;

struct BIT {
    ll c[N];
    int n;

    int lowbit(int x) {
        return x&(-x);
    }
    void modify(int x,ll y) {
        for(; x<=n; x+=lowbit(x)) c[x]+=y;
    }
    ll query(int x) {
        ll ret=0;
        for(; x; x-=lowbit(x)) ret+=c[x];
        return ret;
    }

    ll query(int l,int r) {
        return query(r)-query(l-1);
    }
} bit[20];

int n,m,T;
ll a[N];

int main() {
    scanf("%d%d",&n,&m);
    for(int i=0; i<m; i++) bit[i].n=n;
    for(int i=1; i<=n; i++) {
        scanf("%lld",&a[i]);
        bit[a[i]%m].modify(i,a[i]);
    }
    scanf("%d",&T);
    while(T--) {
        int x,y,z;
        char opt[10];
        scanf("%s%d%d",opt,&x,&y);
        if(opt[0]==+) {
            bit[a[x]%m].modify(x,-a[x]);
            a[x]+=y;
            bit[a[x]%m].modify(x,a[x]);
            printf("%lld\n",a[x]);
        }
        if(opt[0]==-) {
            if(a[x]<y) {
                printf("%lld\n",a[x]);
            } else {
                bit[a[x]%m].modify(x,-a[x]);
                a[x]-=y;
                bit[a[x]%m].modify(x,a[x]);
                printf("%lld\n",a[x]);
            }
        }
        if(opt[0]==s) {
            scanf("%d",&z);
            printf("%lld\n",bit[z].query(x,y));
        }
    }
    return 0;
}

GYM 100741A Queries