NOIAC#71. 作画鬼才 二维差分

NOIAC#71. 作画鬼才

题目传送门

分析

二维差分的板子。
首先一副画的差异度$d_i$是原图差异度$D$去掉那个矩阵再加上那个矩阵的贡献。
一种想法是，按颜色考虑。假设我们计算出了每个位置$i,j$每种颜色$c$的出现次数，计为$mp[c][i][j]$，在此基础之上计算原图的答案。
如果原图$i,j$的颜色是$col[i][j]$，那么在那个位置上的贡献是$val[i][j]={\sum }_{c=0}^{26}|col[i][j]-c|\cdot mp[c][i][j]$
这个时候对$val$取前缀和，就可以快速去掉某个矩阵的贡献。
这个时候再考虑这个矩阵的贡献。如果对每种颜色$c$的$mp[c][i][j]$再取前缀和，就可快速计算出某个矩阵中某种颜色的出现次数。
那么对于那个去掉的矩形，可以采用类似的方法计算贡献。
所以只需要求出$mp[c][i][j]$即可。
这里需要用到二维差分。
设原矩阵为$a_{i,j}$，其二维差分矩阵是${\delta }_{i,j}=a_{i,j}-a_{i,j-1}-a_{i-1,j}+a_{i-1,j-1}$，这样的话修改一个矩阵相当于在其差分矩阵上修改四个位置，最后还原即可。

代码

#include<bits/stdc++.h>
const int N = 1e3 + 10, M = 3e5 + 10;
int ri() {
	char c = getchar(); int x = 0, f = 1; for(;c < '0' || c > '9'; c = getchar()) if(c == '-') f = -1;
	for(;c >= '0' && c <= '9'; c = getchar()) x = (x << 1) + (x << 3) - '0' + c; return x * f;
}
char rc() {
	char c = getchar(); for(;c < 'a' || c > 'z'; c = getchar()) ;
	return c;
}
int ch[N][N], cnt[N][N];
int x[M], y[M], _x[M], _y[M], col[M], n, m, k, s; 
long long sum, mp[30][N][N], val[N][N];
void Add(int x, int y, int _x, int _y, int c) {
	++mp[c][x][y]; ++mp[c][_x + 1][_y + 1];
	--mp[c][x][_y + 1]; --mp[c][_x + 1][y];
}
long long Query(int x, int y, int _x, int _y, int c) {
	long long ret = sum;
	for(int i = 0;i < s; ++i) 
		ret += abs(c - i) * (mp[i][_x][_y] - mp[i][x - 1][_y] - mp[i][_x][y - 1] + mp[i][x - 1][y - 1]);
	ret -= val[_x][_y] - val[x - 1][_y] - val[_x][y - 1] + val[x - 1][y - 1];
	return ret;
}
int main() {
	n = ri(); m = ri(); k = ri(); s = ri();
	for(int i = 1;i <= n; ++i) 
		for(int j = 1;j <= m; ++j)
			ch[i][j] = rc() - 'a';
	for(int i = 1;i <= k; ++i) {
		x[i] = ri(); y[i] = ri(); _x[i] = ri(); _y[i] = ri(); col[i] = rc() - 'a';
		Add(x[i], y[i], _x[i], _y[i], col[i]);
	}
	for(int i = 1;i <= n; ++i)
		for(int j = 1;j <= m; ++j)
			for(int c = 0;c < s; ++c)
				mp[c][i][j] += mp[c][i - 1][j] + mp[c][i][j - 1] - mp[c][i - 1][j - 1],
				cnt[i][j] += mp[c][i][j];
	for(int i = 1;i <= n; ++i)
		for(int j = 1;j <= m; ++j)
			mp[ch[i][j]][i][j] += k - cnt[i][j];
	for(int i = 1;i <= n; ++i)
		for(int j = 1;j <= m; ++j) {
			for(int c = 0;c < s; ++c)
				val[i][j] += abs(c - ch[i][j]) * mp[c][i][j];
			sum += val[i][j];
		}
	for(int i = 1;i <= n; ++i)
		for(int j = 1;j <= m; ++j) {
			for(int c = 0;c < s; ++c)
				mp[c][i][j] += mp[c][i - 1][j] + mp[c][i][j - 1] - mp[c][i - 1][j - 1];
			val[i][j] += val[i - 1][j] + val[i][j - 1] - val[i - 1][j - 1];
		}
	long long ans = 1e18; int id;
	for(int i = 1;i <= k; ++i) {
		long long t = Query(x[i], y[i], _x[i], _y[i], col[i]);
		if(ans > t) 
			ans = t, id = i;
	}
	printf("%lld %d\n", ans, id);
	return 0;
}