HDU 5671 矩阵的交换行、交换列,整行加一个数 思维题

点击打开链接

对于交换行、交换列的操作,分别记录当前状态下每一行、每一列是原始数组的哪一行、哪一列即可。

对每一行、每一列加一个数的操作,也可以两个数组分别记录。注意当交换行、列的同时,也要交换增量数组。

输出时通过索引找到原矩阵中的值,再加上行、列的增量。

复杂度O(q+mn)O(q+mn)O(q+mn)

#include<iostream>
#include<cstdio>
#include<cstring>

using namespace std;
int a[1010][1010];
int sx[1010],sy[1010],tx[1010],ty[1010];
long long xm[1001],ym[1010];
int main()
{
    int T;
    while(~scanf("%d",&T))
    {

        while(T--)
        {
            int n,m,q;
            scanf("%d%d%d",&n,&m,&q);

            for(int i=1;i<=n;i++)
                for(int j=1;j<=m;j++)
                {
                    scanf("%d",&a[i][j]);
                }

            for(int i=1;i<=n;i++) sx[i]=i,tx[i]=i,xm[i]=0;
            for(int j=1;j<=m;j++) sy[j]=j,ty[j]=j,ym[j]=0;


            for(int i=0;i<q;i++)
            {
                int t1,t2,t3;
                scanf("%d%d%d",&t1,&t2,&t3);
                if(t1==1)
                {
                   swap(sx[t2],sx[t3]);
     
                }
                else if(t1==2)
                {
                   swap(sy[t2],sy[t3]);
                }
                else if(t1==3)
                {
                   xm[sx[t2]]+=t3;
                }
                else
                {
                    ym[sy[t2]]+=t3;
                }
            }
            
            for(int i=1;i<=n;i++)
            {
                for(int j=1;j<=m;j++)
                {
                    if(j!=1)
                        printf(" ");
                    printf("%lld",(long long)a[sx[i]][sy[j]]+xm[sx[i]]+ym[sy[j]]);
                }
                printf("\n");
            }
        }

    }
    return 0;
}


你可能感兴趣的:(HDU 5671 矩阵的交换行、交换列,整行加一个数 思维题)