Matrix and Shifts（矩阵平移、二维前缀和）

一、题意

仅有0和1组成的矩阵，矩阵可以整体性地上下移动或左右移动，移动不花费代价，然后可以通过异或运算，将矩阵转换成对角线元素均为1，而其他值为0的矩阵。每次异或运算代价1，求最小代价

二、思路

1、对于n*n矩阵的平移，可以在读入的时候，把它变成2n*2n的矩阵（平移至右、下、右下），新矩阵会包括所有的平移情况

2、暴力求出新矩阵的二维前缀和（（1,1）到（i，j）矩阵中1的个数，每条对角线1的个数）

sum[i][j] = sum[i-1][j]+sum[i][j-1]-sum[i-1][j-1]+s[i][j]-'0';
p[i][j] = p[i-1][j-1]+s[i][j]-'0';

3、每个n*n矩阵的代价为：非对角线需要修改的1和对角线需要修改的0

        即（所有1的个数-对角线上1的个数）+（n-对角线上1的个数）

三、代码

#include<bits/stdc++.h>
using namespace std;
char s[4001][4001];
int sum[4001][4001];
int p[4001][4001];
int main()
{
	int t,n;
	scanf("%d",&t);
	while(t--)
	{
		int ans=0x3f3f3f3f;
		scanf("%d",&n);
		getchar();
		for(int i=1; i<=2*n; ++i)
			for(int j=1; j<=2*n; ++j)
				sum[i][j] = 0;
		for(int i=1; i<=n; ++i)
		{
			for(int j=1; j<=n; ++j)
				s[i][j] = s[n+i][j] = s[i][n+j] = s[n+i][n+j] = getchar();
			getchar();
		}
		for(int i=1; i<=2*n; ++i)
			for(int j=1; j<=2*n; ++j)
			{
				sum[i][j] = sum[i-1][j]+sum[i][j-1]-sum[i-1][j-1]+s[i][j]-'0';
				p[i][j] = p[i-1][j-1]+s[i][j]-'0';
			}
		for(int i=n; i<=2*n; ++i)
			for(int j=n; j<=2*n; ++j)
			{
				int cnt = p[i][j]-p[i-n][j-n];
				int num = sum[i][j]-sum[i][j-n]-sum[i-n][j]+sum[i-n][j-n];
				ans = min(ans, num-cnt+n-cnt);
			}
		printf("%d\n",ans);
	}
	return 0;
}