poj3467(维护前缀长度)

链接：点击打开链接

题意：给定一个n*m的矩阵，每个点有一个权值，现在有两种操作，一种是将其中任何一个元素改成1~c中的任意值，另一种是输出权值u按十字计数的种数(具体看样例)

代码：

#include <queue>
#include <vector>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <iostream>
#include <algorithm>
using namespace std;
int n,m;
int num[105],s[105][105];
int le[105][105],re[105][105],up[105][105],dw[105][105];
void cal(int x,int y,int id){
    int i;
    if(id==1){
        for(i=x+1;i<=n;i++){
            if(s[i][y]==s[x][y])
            dw[x][y]++;
            else
            break;
        }
    }
    else if(id==2){
        for(i=x-1;i>=1;i--){
            if(s[i][y]==s[x][y])
            up[x][y]++;
            else
            break;
        }
    }
    else if(id==3){
        for(i=y-1;i>=1;i--){
            if(s[x][i]==s[x][y])
            le[x][y]++;
            else
            break;
        }
    }
    else{
        for(i=y+1;i<=m;i++){
            if(s[x][i]==s[x][y])
            re[x][y]++;
            else
            break;
        }
    }
}
void update(int x,int y,int op){                //每个点分三种情况讨论一下即可
    int i;
    if(s[x][y]==op)
    return;
    for(i=1;i<=x-1;i++){
        if(dw[i][y]<x-i-1)
        continue;
        else if(dw[i][y]==x-i-1){
            if(s[i][y]==op){
                dw[i][y]++;
                if(x+1<=n&&s[x+1][y]==s[i][y])
                dw[i][y]+=(dw[x+1][y]+1);
            }
        }
        else
        dw[i][y]=x-i-1;
    }
    for(i=n;i>=x+1;i--){
        if(up[i][y]<i-x-1)
        continue;
        else if(up[i][y]==i-x-1){
            if(s[i][y]==op){
                up[i][y]++;
                if(x-1>=1&&s[x-1][y]==s[i][y])
                up[i][y]+=(up[x-1][y]+1);
            }
        }
        else
        up[i][y]=i-x-1;
    }
    for(i=1;i<=y-1;i++){
        if(re[x][i]<y-i-1)
        continue;
        else if(re[x][i]==y-i-1){
            if(s[x][i]==op){
                re[x][i]++;
                if(y+1<=m&&s[x][y+1]==s[x][i])
                re[x][i]+=(re[x][y+1]+1);
            }
        }
        else
        re[x][i]=y-i-1;
    }
    for(i=m;i>=y+1;i--){
        if(le[x][i]<i-y-1)
        continue;
        else if(le[x][i]==i-y-1){
            if(s[x][i]==op){
                le[x][i]++;
                if(y-1>=1&&s[x][y-1]==s[x][i])
                le[x][i]+=(le[x][y-1]+1);
            }
        }
        else
        le[x][i]=i-y-1;
    }
    s[x][y]=op;
    up[x][y]=dw[x][y]=le[x][y]=re[x][y]=0;
    for(i=1;i<=4;i++)
    cal(x,y,i);
}
int main(){
    char ch;
    int c,q,i,j,u,v,id,op;
    while(scanf("%d%d%d%d",&n,&m,&c,&q)!=EOF){
        for(i=1;i<=n;i++)                       //直接维护每个点向四个方向的连续长度
        for(j=1;j<=m;j++)                       //当改任何一个值时，可以每个点直接O(1)
        scanf("%d",&s[i][j]);                   //进行更改，因此复杂度就为O(Q*2*(n+m))
        memset(le,0,sizeof(le));
        memset(re,0,sizeof(re));
        memset(up,0,sizeof(up));
        memset(dw,0,sizeof(dw));
        memset(num,0,sizeof(num));
        for(i=1;i<=n;i++){                      //直接维护出每个点四个方向的值
            for(j=1;j<=m;j++){
                for(id=1;id<=4;id++)
                cal(i,j,id);
            }
        }
        for(i=1;i<=n;i++)
        for(j=1;j<=m;j++)                       //四个方向最小值即为每个点的个数
        num[s[i][j]]+=min(min(up[i][j],dw[i][j]),min(le[i][j],re[i][j]));
        while(q--){
            cin>>ch;
            if(ch=='Q'){
                scanf("%d",&u);
                printf("%d\n",num[u]);
            }
            else{
                scanf("%d%d%d",&u,&v,&op);
                for(j=1;j<=m;j++)               //询问时可以先将原来的值都减掉
                num[s[u][j]]-=min(min(up[u][j],dw[u][j]),min(le[u][j],re[u][j]));
                for(i=1;i<=n;i++){              //当更新后再重新加回来
                    if(i==u)
                    continue;
                    num[s[i][v]]-=min(min(up[i][v],dw[i][v]),min(le[i][v],re[i][v]));
                }
                update(u,v,op);
                for(j=1;j<=m;j++)
                num[s[u][j]]+=min(min(up[u][j],dw[u][j]),min(le[u][j],re[u][j]));
                for(i=1;i<=n;i++){
                    if(i==u)
                    continue;
                    num[s[i][v]]+=min(min(up[i][v],dw[i][v]),min(le[i][v],re[i][v]));
                }
            }
        }
    }
    return 0;
}