zcmu 1725 关于矩阵（前缀和）

【题目】

1725: 关于矩阵

Time Limit: 1 Sec  Memory Limit: 128 MB
Submit: 284  Solved: 114
[Submit][Status][Web Board]

Description

一个N*M的矩阵A(| A[i][j] |<=10^3)，有3中操作

1. add i,j,val :  把矩阵A[i][j]位置加上val(|val|<=10^3)

2. sub i,j val:  把矩阵A[i][j]位置减去val(|val|<=10^3)

3.sum x1,y1,x2,y2: 求以(x1,y1),(x2,y2)为对角线的子矩阵和(1<=x1<=x2<=N,1<=y1<=y2<=M)

Input

第一行N,M(1<=N,M<=1000)表示矩阵的大小

然后N*M行矩阵的初始值

接下一个Q(1<=Q<=100000)表示操作

然后Q行，每一行如上所述

Output

一个数表示答案

Sample Input

2 2
1 2
3 4
5
add 1 1 2
sum 1 1 1 1
sub 1 1 2
sum 1 1 1 1
sum 1 1 2 2

Sample Output

3
1
10

【题解】

一维前缀和

【代码】

#include<bits/stdc++.h>
using namespace std;
int a[1005][1005];
main()
{
    int n,m,q,i,j,k;
    while(~scanf("%d%d",&n,&m))
    {
        memset(a,0,sizeof(a));
        for(i=1;i<=n;i++)
            for(j=1;j<=m;j++)
            {
                scanf("%d",&a[i][j]);
                a[i][j]+=a[i][j-1];
            }
        scanf("%d",&q);
        char s[5];
        int x,y,xx,yy,v;
        for(k=0;k<q;k++)
        {
            scanf("%s%d%d",s,&x,&y);
            if(s[2]=='d')
            {
                scanf("%d",&v);
                for(i=y;i<=m;i++)
                    a[x][i]+=v;
            }
            else if(s[2]=='b')
            {
                scanf("%d",&v);
                for(i=y;i<=m;i++)
                    a[x][i]-=v;
            }
            else
            {
                scanf("%d%d",&xx,&yy);
                int sum=0;
                for(i=x;i<=xx;i++)
                    sum+=a[i][yy]-a[i][y-1];
                printf("%d\n",sum);
            }
        }
    }
}