Gym - 100851A Adjustment Office（O（1）求行列和）

Adjustment Office

Gym - 100851A

2       3       4

3       4       5

4       5       6

n<=10^6，q<=10^5

每个坐标点的值为x+y。

q次查询，每次求某行或某列的和，然后清空。

in                   out

3 7                 12
R 2                 10
C 3                 0
R 2                 5
R 1                 5
C 2                 4
C 1                 0
R 3

#include <bits/stdc++.h>
#define MAX 2000005
typedef long long ll;
using namespace std;

int bh[MAX],bl[MAX];
ll pre[MAX];
ll h[MAX],l[MAX];
char s[];
ll vh=,vl=,vhc=,vlc=;

void init(){
    pre[]=;
    for(int i=;i<=;i++){
        pre[i]=pre[i-]+i;
    }
}
int main(void)
{
    freopen("adjustment.in","r",stdin);
    freopen("adjustment.out","w",stdout);
    init();
    int n,q;
    ll x;
    scanf("%d%d",&n,&q);
    for(int i=;i<=n;i++){
        h[i]=pre[i+n]-pre[i];
        l[i]=pre[i+n]-pre[i];
    }
    while(q--){
        scanf(" %s %I64d",s,&x);
        if(s[]=='R'){
            if(bh[x]==){
                printf("0\n");
                continue;
            }
            bh[x]=;
            vh+=x;
            vhc++;
            ll ans=h[x];
            ans-=vlc*x+vl;
            printf("%I64d\n",ans);
        }
        else{
            if(bl[x]==){
                printf("0\n");
                continue;
            }
            bl[x]=;
            vl+=x;
            vlc++;
            ll ans=l[x];
            ans-=vhc*x+vh;
            printf("%I64d\n",ans);
        }
    }
    return ;
}