CF1982D Beauty of the mountains

【题意】

Nikita 喜欢爬山。当地的山可以看作一个由 $(n\times m)$ 个格子组成的 $n\times m$ 的矩形，每一个格子都有一个初始非负高度 $a_{i,j}$。

格子有两种类型：

• 有雪
• 无雪

Nikita 有一种超能力：他可以任意选择一个 $k\times k$ 的子矩形，使这个子矩形中的所有格子的高度增加同一个常量 $c$（$c$ 可以自己指定，每次操作未必相同）。$c$ 可以是负数。

Nikita 希望通过有限次操作使得有雪的格子的总高度和无雪的格子的总高度相等。

你的任务是判断是否可行。

【读入】

本题有多组数据。

第一行一个整数 $t(1\le t\le 1\times 10^4)$，表示一共有 $t$ 组数据。

每组数据的第一行三个整数 n , m , k ( 1 ≤ n , m ≤ 500 , 1 ≤ k ≤ min ⁡ { n , m } ) n,m,k(1 \leq n,m \leq 500, 1\leq k \leq \min \left \{ n,m \right \}) n,m,k(1≤n,m≤500,1≤k≤min{n,m})。

接下来 $n$ 行，每行 $m$ 个整数，表示初始高度 $a_{i,j}(0\le a_{i,j}\le 1\times 10^9)$。

接下来 $n$ 行，每行一个长度为 $m$ 的有 $0,1$ 构成的字符串（没有空格），$0$ 表示该格子无雪，$1$ 表示有雪。

保证所有数据的 $n\times m$ 总和不超过 $250,000$。

【输出】

$t$ 行，表示答案。如果 Nikita 可以在有限次操作后达成目标，则输出 Yes，否则输出 No。

不区分大小写，如 YES，Yes，yEs 都表示给出肯定答案。

Translated by HPXXZYY.

---

$\text{[Solution]}$

显然，每个格子的值具体是多少我们并不关心，我们只关心两种格子的值的差值，记为 $\Delta$。

很显然，对 $k\times k$ 的子矩形中的每一个数都加上 $c_1$，再加上 $c_2$，与直接加上 $c=c_1+c_2$ 没有区别。因此，如果有解，一定存在这样的解：我们对每个需要变化的子矩形最多进行 $1$ 次操作。

当然，如果对某个子矩形不进行操作，我们也可以对它进行一次加上 $c=0$ 的操作。这样，一定存在这样的一组解：我们对每个 $k\times k$ 的子矩形都进行且只进行 $1$ 次操作。

对于某个子矩形，加入它有 $x$ 个有雪的格子，那么它一定有 $\left(k^2-x\right)$ 个没有雪的格子。那么对这个子矩形进行一次操作后，会使 $\Delta$ 变化为 $\Delta +c\times \left[\left(k^2-x\right)-x\right]$（显然你想要 $\Delta$ 减小只需让 $c$ 为负数）。

对每个子矩形维护这样一个特征值 $\left[\left(k^2-x\right)-x\right]$，记为 $t$。对这个子矩形的操作必然使 $\Delta$ 增加或减小 $t$ 的倍数。

这样我们就获得了一系列的 $t$，记为 $t_1,t_2,t_3,\dots ,t_g$。问题变成解下面的不定方程是否有整数解：

$$\Delta =\sum _{i=1}^g{c}_i\cdot t_i$$

显然如果 $\Delta$ 是 gcd ⁡ { t i } \gcd \{ t_{i} \} gcd{ti​} 的倍数，则有解；反之则无解。

$\text{[code]}$

typedef long long ll;

char c[510];
int n,m,T,res,k;ll delta;
int a[510][510],b[510][510],s[510][510];

inline int Abs(int u){
	return ((u<0)?-u:u);
}
inline int gcd(int u,int v){
	if (v==0) return u;
	return gcd(v,u%v);
}

int main(){
	T=read();
	while (T--){
		n=read();m=read();k=read();
		for(int i=1;i<=n;i++)
			for(int j=1;j<=m;j++)
				a[i][j]=read();
		
		for(int i=1;i<=n;i++)
			for(int j=1;j<=m;j++)
				b[i][j]=s[i][j]=0;//只能这么清空 
		delta=0;res=-1;
		
		for(int i=1;i<=n;i++){
			scanf("%s",c+1);
			for(int j=1;j<=m;j++)
				if (c[j]=='1'){
					delta+=a[i][j];
					b[i][j]=1;
				}
				else{
					delta-=a[i][j];
					b[i][j]=-1;
				}
		}
		
		for(int i=1;i<=n;i++)
			for(int j=1;j<=m;j++)
				s[i][j]=s[i-1][j]+s[i][j-1]-s[i-1][j-1]+b[i][j];
		
		for(int i=k;i<=n;i++)
			for(int j=k;j<=m;j++){
				int tmp=s[i][j]-s[i-k][j]-s[i][j-k]+s[i-k][j-k];
				if (res==-1) res=Abs(tmp);
				else res=gcd(res,Abs(tmp));
			}
		
		if (res==0) printf("%s\n",(delta==0?"Yes":"NO"));
		else if (delta%res==0) printf("yEs\n");
		else printf("nO\n");
	}
	
	return 0;
}