QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#352934#7989. 三染色ucup-team1004TL 229ms53456kbC++142.4kb2024-03-13 18:29:492024-03-13 18:29:55

Judging History

你现在查看的是最新测评结果

  • [2024-03-13 18:29:55]
  • 评测
  • 测评结果:TL
  • 用时:229ms
  • 内存:53456kb
  • [2024-03-13 18:29:49]
  • 提交

answer

#include<bits/stdc++.h>
using namespace std;
using ll=long long;
#define all(a) (a).begin(),(a).end()
#ifdef DEBUG
template<class T>
ostream& operator << (ostream &out,vector<T> a){
	out<<'[';
	for(T x:a)out<<x<<',';
	return out<<']';
}
template<class T>
vector<T> ary(T *a,int l,int r){
	return vector<T>{a+l,a+1+r};
}
template<class T>
void debug(T x){
	cerr<<x<<endl;
}
template<class T,class...S>
void debug(T x,S...y){
	cerr<<x<<' ',debug(y...);
}
#else
#define debug(...) void()
#endif
const int N=5,M=51,K=pow(3,N)+3,mod=998244353;
const int w[3][3]={{0,-1,1},{1,0,-1},{-1,1,0}};
int n,m,a[N][M];
int v[K][N],is[M][K],ok[K][K],mx[K],cur[K];
bool chk(int a,int b,int c,int d){
	return !((1<<a|1<<b|1<<c|1<<d)==7&&(a==b||b==c||c==d||d==a));
}
int f[M][K][M+N][M+N];
void add(int &x,int y){
	(x+=y)>=mod&&(x-=mod);
}
int main(){
	cin>>n>>m;
	for(int i=0;i<n;i++){
		for(int j=1;j<=m;j++)cin>>a[i][j];
	}
	int U=pow(3,n)-0.5;
	for(int S=0;S<=U;S++){
		for(int i=0,x=S;i<n;i++){
			v[S][i]=x%3,x/=3;
		}
	}
	for(int j=1;j<=m;j++){
		for(int S=0;S<=U;S++){
			is[j][S]=1;
			for(int i=0;i<n;i++){
				if(a[i][j]!=3&&a[i][j]!=v[S][i])is[j][S]=0;
			}
		}
	}
	for(int S=0;S<=U;S++){
		mx[S]=cur[S]=0;
		for(int i=1,x=0;i<n;i++){
			x+=w[v[S][i-1]][v[S][i]];
			if(x>mx[S])mx[S]=x,cur[S]=i;
		}
	}
	for(int S=0;S<=U;S++){
		for(int T=0;T<=U;T++){
			ok[S][T]=1;
			for(int i=0;i+1<n;i++){
				if(!chk(v[S][i],v[T][i],v[T][i+1],v[S][i+1]))ok[S][T]=0;
			}
		}
	}
	for(int S=0;S<=U;S++)if(is[1][S]){
		f[1][S][0][cur[S]]=1;
	}
	for(int i=1;i<m;i++){
		int lim=i-1+n;
		for(int S=0;S<=U;S++)if(is[i][S]){
			vector<int>pos;
			for(int T=0;T<=U;T++)if(is[i+1][T]&&ok[S][T]){
				pos.push_back(T);
			}
			for(int p=0;p<lim;p++){
				for(int d=0;d<lim;d++){
					if(!f[i][S][p][d])continue;
					for(int T:pos){
						int q=p+mx[S]-w[v[S][0]][v[T][0]]-mx[T];
						if(q>0){
							add(f[i+1][T][q][d],f[i][S][p][d]);
						}else if(q==0){
							add(f[i+1][T][q][min(d,i+cur[T])],f[i][S][p][d]);
						}else{
							add(f[i+1][T][0][i+cur[T]],f[i][S][p][d]);
						}
					}
				}
			}
		}
	}
	int ans1=0,ans2=0;
	for(int S=0;S<=U;S++){
		int lim=n+m-1;
		for(int p=0;p<lim;p++){
			for(int d=0;d<lim;d++){
				add(ans1,f[m][S][p][d]);
				ans2=(ans2+1ll*d*f[m][S][p][d])%mod;
			}
		}
	}
	cout<<ans1<<' '<<ans2<<endl;
	return 0;
}

Details

Tip: Click on the bar to expand more detailed information

Test #1:

score: 100
Accepted
time: 1ms
memory: 3908kb

input:

2 2
1 0
3 2

output:

1 2

result:

ok single line: '1 2'

Test #2:

score: 0
Accepted
time: 0ms
memory: 7256kb

input:

5 5
3 3 3 3 2
2 3 3 3 1
1 3 3 3 3
3 3 3 3 3
3 3 3 3 3

output:

50830224 170059345

result:

ok single line: '50830224 170059345'

Test #3:

score: 0
Accepted
time: 229ms
memory: 53456kb

input:

5 50
3 3 3 3 3 3 3 3 1 0 3 3 3 1 3 3 3 3 3 3 3 3 3 3 2 3 3 3 3 2 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3
3 3 3 1 3 3 1 3 3 1 3 3 3 3 3 3 3 3 3 1 3 3 3 3 2 3 3 3 3 2 0 3 0 3 3 3 0 3 3 3 3 3 3 3 0 0 3 3 3 3
1 3 3 3 3 3 2 3 3 3 3 3 3 3 3 3 3 3 3 3 1 3 3 1 3 3 3 0 1 3 3 3 0 3 3 3 3 3 2 3 3 1 3 3 0 3 3 3...

output:

988900801 3995091

result:

ok single line: '988900801 3995091'

Test #4:

score: -100
Time Limit Exceeded

input:

5 50
3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3
3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3
3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3...

output:


result: