10 pts 求调
查看原帖
10 pts 求调
901195
__Luna__楼主2024/11/28 18:31
#include<bits/stdc++.h>
#define int long long
const int mod=998244353;
using namespace std;
char a[1005][1005];
int b[1005][1005],c[1005][1005],d[1005][1005];
void sv()
{
	memset(a,0,sizeof a);
	memset(b,0,sizeof b);
	memset(c,0,sizeof c);
	memset(d,0,sizeof d);
	int n,m,C,F,ac=0,af=0;
	cin>>n>>m>>C>>F;
	for(int i=0;i<n;i++)
	{
		for(int j=0;j<m;j++)
		{
			cin>>a[i][j];
		}
	}
	for(int i=0;i<n;i++)
	{
		for(int j=m-1;j>=0;j--)
		{
			if(a[i][j]=='1')b[i][j]=0;
			else b[i][j]=b[i][j+1]+1;
		}
	}
	for(int i=0;i<m;i++)
	{
		if(b[0][i])c[0][i]=b[0][i]-1;
		else c[0][i]=0;
	}
	for(int i=1;i<n;i++)
	{
		for(int j=0;j<m;j++)
		{
			if(b[i][j])c[i][j]=c[i-1][j]+b[i][j]-1;
			else c[0][i]=0;
		}
	}
	for(int i=n-1;i>=0;i--)
	{
		for(int j=0;j<m;j++)
		{
			if(a[i][j]=='1')d[i][j]=0;
			else d[i][j]=d[i+1][j]+1;
		}
	}
	for(int j=0;j<m;j++)
	{
		for(int i=2;i<n;i++)
		{
			if(b[i][j]>1&&b[i-1][j]!=0)ac=(ac+((b[i][j]-1)*c[i-2][j])%mod)%mod,af=(af+(((b[i][j]-1)*c[i-2][j])%mod*(d[i][j]-1))%mod)%mod;
		}
	}
	cout<<(ac*C)%mod<<" "<<(af*F)%mod<<endl;
}
signed main()
{
	int t,_;
	cin>>t>>_;
	while(t--)sv();
}

样例都过了,谴责大样例。

2024/11/28 18:31
加载中...