【题目泛做】雅礼集训：attack（容斥）（单调栈）（差分）

---

题解：

首先考虑一个简单容斥：

对于一个一维上区间，显然它的大小为1的子区间个数永远是大小为2的子区间个数+1。

拓展到二维上，显然一个矩阵的$2\ast 2$的子矩阵个数$+1\ast 1$子矩阵$-2\ast 1$子矩阵$-1\ast 2$子矩阵。

所以考虑对于所有2*2以内的小矩阵统计答案。

显然对于一个矩阵，答案为包含它的全$1$矩阵的个数的$k$次方。

我们需要对于一个小矩阵统计有多少个矩形完全包含它。

一个显然的想法是对于每一个矩形，做一个差分数组，然后求一遍二维前缀和。

显然我们不能枚举所有矩形。

考虑利用单调栈算出以每个点为顶角的矩形数量，可以直接得到各个位置应该放的差分标记数量。

想清楚了之后就剩下模拟了。

---

代码：

#include<bits/stdc++.h>
#define ll long long
#define re register
#define cs const

cs int mod=998244353;
inline int add(int a,int b){a+=b-mod;return a+(a>>31&mod);}
inline int dec(int a,int b){a-=b;return a+(a>>31&mod);}
inline int mul(int a,int b){ll r=(ll)a*b;return r>=mod?r%mod:r;}
inline int power(int a,int b,int res=1){
	for(;b;b>>=1,a=mul(a,a))(b&1)&&(res=mul(res,a));
	return res;
}
inline void Inc(int &a,int b){a+=b-mod;a+=a>>31&mod;}
inline void Dec(int &a,int b){a-=b;a+=a>>31&mod;}

cs int N=2e3+7;

int n,m,k,ans;

template<class T>
inline void rev_x(T A[][N]){
	for(int re i=1;i<=n/2;++i)
	for(int re j=1;j<=m;++j)std::swap(A[i][j],A[n-i+1][j]);
}

template<class T>
inline void rev_y(T A[][N]){
	for(int re i=1;i<=n;++i)
	for(int re j=1;j<=m/2;++j)std::swap(A[i][j],A[i][m-j+1]);
}

char mp[N][N];
int s0[N][N],s1[N][N],s2[N][N],s3[N][N];
int s[N][N];

inline void solve(int s[][N]){
	static int h[N][N];
	for(int re i=1;i<=n;++i)
	for(int re j=1;j<=m;++j)h[i][j]=(mp[i][j]=='0')?0:h[i-1][j]+1;
	static int siz[N],ht[N],sum,t;
	for(int re i=1;i<=n;++i){
		sum=t=0;
		for(int re j=1;j<=m;++j){
			int now_siz=1;
			while(t&&ht[t]>=h[i][j]){
				now_siz+=siz[t];
				Dec(sum,mul(siz[t],ht[t]));
				--t;
			}
			ht[++t]=h[i][j];siz[t]=now_siz;
			Inc(sum,mul(siz[t],ht[t]));
			s[i][j]=sum;
		}
	}
}

inline void calc(int coef){
	int tmp=0;
	for(int re i=1;i<=n;++i)
	for(int re j=1;j<=m;++j){
		Dec(s[i][j],s[i-1][j-1]);
		Inc(s[i][j],s[i-1][j]);
		Inc(s[i][j],s[i][j-1]);
		Inc(tmp,power(s[i][j],k));
	}
	coef==1?Inc(ans,tmp):Dec(ans,tmp);
	for(int re i=1;i<=n;++i)memset(s[i]+1,0,sizeof(int)*m);
}

signed main(){
#ifdef zxyoi
	freopen("atk.in","r",stdin);
#endif
	scanf("%d%d%d",&n,&m,&k);
	for(int re i=1;i<=n;++i)scanf("%s",mp[i]+1);
	solve(s3);
	rev_x(mp);solve(s1);rev_x(s1);
	rev_y(mp);solve(s0);rev_x(s0),rev_y(s0);
	rev_x(mp);solve(s2);rev_y(s2);
	for(int re i=1;i<=n;++i)
	for(int re j=1;j<=m;++j){
		Inc(s[i][j],s0[i][j]);
		Inc(s[i+1][j+1],s3[i][j]);
		Dec(s[i][j+1],s1[i][j]);
		Dec(s[i+1][j],s2[i][j]);
	}calc(1);
	for(int re i=1;i<=n;++i)
	for(int re j=1;j<=m;++j){
		Inc(s[i][j],s0[i][j]);
		Inc(s[i+1][j],s3[i][j]);
		Dec(s[i][j],s1[i][j]);
		Dec(s[i+1][j],s2[i][j]);
	}calc(-1);
	for(int re i=1;i<=n;++i)
	for(int re j=1;j<=m;++j){
		Inc(s[i][j],s0[i][j]);
		Inc(s[i][j+1],s3[i][j]);
		Dec(s[i][j+1],s1[i][j]);
		Dec(s[i][j],s2[i][j]);
	}calc(-1);
	for(int re i=1;i<=n;++i)
	for(int re j=1;j<=m;++j){
		Inc(s[i][j],s0[i][j]);
		Inc(s[i][j],s3[i][j]);
		Dec(s[i][j],s1[i][j]);
		Dec(s[i][j],s2[i][j]);
	}calc(1);
	std::cout<<ans<<"\n"; 
	return 0;
}