「雅礼集训 2018 Day11」进攻！[容斥+单调栈]

传送门
考虑枚举最后的相交的矩阵，统计最后相交出来为枚举的那个矩阵的答案就可以不重不漏
考虑到那个矩阵内 1 * 1 的个数 - 1 * 2 的个数 - 2 * 1 的个数 + 2 * 2的个数是等于 1 的
也就是说我们可以对 1 * 1 的统计一遍答案，对 1 * 2 的统计一遍 … 就可以容斥了
统计答案需要知道包涵一个点的矩阵个数，可以差分
暴力做法就是枚举每个合法矩阵，然后差分
但发现我们可以对于所有矩阵一起差分，需要统计以一个点为左下角的矩阵个数
用单调栈即可求出

#include<bits/stdc++.h>
#define N 2050
using namespace std;
const int Mod = 998244353;
typedef long long ll;
ll add(ll a, ll b){ return (a + b) % Mod;}
ll mul(ll a, ll b){ return (a * b) % Mod;}
void Add(ll &a, ll b){ a = add(a, b);}
ll power(ll a, ll b){ ll ans = 1;
	for(;b;b>>=1){ if(b&1) ans = mul(ans, a); a = mul(a, a);}
	return ans;
}
int n, m, k; ll ans;
char mp[N][N];
ll s0[N][N], s1[N][N], s2[N][N], s3[N][N], s[N][N];
int h[N][N], hi[N], siz[N], top;
template<class T>
void revx(T a[N][N]){for(int i = 1; i <= n/2; i++) for(int j = 1; j <= m; j++) swap(a[i][j], a[n-i+1][j]);}
template<class T>
void revy(T a[N][N]){for(int i = 1; i <= n; i++) for(int j = 1; j <= m/2; j++) swap(a[i][j], a[i][m-j+1]);}

void calc(ll s[N][N]){
	memset(h, 0, sizeof(h));
	for(int i = 1; i <= n; i++)
		for(int j = 1; j <= m; j++)
			h[i][j] = (mp[i][j] == '0') ? 0 : h[i-1][j] + 1;
	memset(hi, 0, sizeof(hi)); 
	for(int i = 1; i <= n; i++){
		ll sum = 0; top = 0;
		for(int j = 1; j <= m; j++){
			int now = 1;
			while(top && hi[top] >= h[i][j]){
				now += siz[top];
				Add(sum, Mod - mul(siz[top], hi[top]));
				top--;
			}
			hi[++top] = h[i][j]; siz[top] = now;
			Add(sum, mul(siz[top], hi[top]));
			s[i][j] = sum;
		}
	}
}
void solve(int typ){
	ll tmp = 0;
	for(int i = 1; i <= n; i++){
		for(int j = 1; j <= m; j++){
			Add(s[i][j], Mod - s[i-1][j-1]);
			Add(s[i][j], s[i-1][j]);
			Add(s[i][j], s[i][j-1]);
			Add(tmp, power(s[i][j], k));
		}
	} typ == 1 ? Add(ans, tmp) : Add(ans, Mod - tmp);
	memset(s, 0, sizeof(s));
}
int main(){
	scanf("%d%d%d", &n, &m, &k);
	for(int i = 1; i <= n; i++) scanf("%s", mp[i] + 1);
	calc(s3);
	revx(mp); calc(s1); revx(s1);
	revy(mp); calc(s0); revx(s0); revy(s0);
	revx(mp); calc(s2); revy(s2);
	for(int i = 1; i <= n; i++){
		for(int j = 1; j <= m; j++){
			Add(s[i][j], s0[i][j]);
			Add(s[i+1][j+1], s3[i][j]);
			Add(s[i][j+1], Mod-s1[i][j]);
			Add(s[i+1][j], Mod-s2[i][j]);
		}
	} 
	solve(1);
	for(int i = 1; i <= n; i++){
		for(int j = 1; j <= m; j++){
			Add(s[i][j], s0[i][j]);
			Add(s[i+1][j], s3[i][j]);
			Add(s[i][j], Mod-s1[i][j]);
			Add(s[i+1][j], Mod-s2[i][j]);
		}
	} solve(-1);
	for(int i = 1; i <= n; i++){
		for(int j = 1; j <= m; j++){
			Add(s[i][j], s0[i][j]);
			Add(s[i][j+1], s3[i][j]);
			Add(s[i][j+1], Mod-s1[i][j]);
			Add(s[i][j], Mod-s2[i][j]);
		}
	} solve(-1);
	for(int i = 1; i <= n; i++){
		for(int j = 1; j <= m; j++){
			Add(s[i][j], s0[i][j]);
			Add(s[i][j], s3[i][j]);
			Add(s[i][j], Mod-s1[i][j]);
			Add(s[i][j], Mod-s2[i][j]);
		}
	} solve(1);
	cout << ans; return 0;
}