记忆化搜索求调
查看原帖
记忆化搜索求调
520748
_Ch1F4N_楼主2023/8/28 10:31

如题,第 33 个样例过不去:

#include<bits/stdc++.h>
#define int long long
//#define lowbit(x) (x&-(x))
using namespace std;
//const int maxn =
const int mod = 998244353;
vector<int> dight;
int dp[65][10][10][10][2][2][2][2][2][2];
int a1,a2,a3;
int solve(int p,int r1,int r2,int r3,bool mx1,bool mx2,bool mx3,bool c1,bool c2,bool c3){//当前考虑到第 p 位且 x1,x2,x3 模 a1,a2,a3 为 r1,r2,r3 是否到达上界 是否全部填 0
	if(dp[p][r1][r2][r3][mx1][mx2][mx3][c1][c2][c3]!=-1) return dp[p][r1][r2][r3][mx1][mx2][mx3][c1][c2][c3];
    if(p==0){ 
		if(r1==0&&r2==0&&r3==0&&c1==false&&c2==false&&c3==false){
			return dp[p][r1][r2][r3][mx1][mx2][mx3][c1][c2][c3]=1;
		}
		else{
			return dp[p][r1][r2][r3][mx1][mx2][mx3][c1][c2][c3]=0;
		}
	}
	int res=0;
	for(int i=0;i<=(mx1==true?dight[p-1]:1);i++){
		for(int j=0;j<=(mx2==true?dight[p-1]:1);j++){
			for(int k=0;k<=(mx3==true?dight[p-1]:1);k++){
				if((i^j)^k==0){
					res+=solve(p-1,(a1+r1+((1<<(p-1))*i)%a1)%a1,(a2+r2+((1<<(p-1))*j)%a2)%a2,(a3+r3+((1<<(p-1))*k)%a3)%a3,mx1&(i==dight[p-1]),mx2&(j==dight[p-1]),mx3&(k==dight[p-1]),c1&&(i==0),c2&&(j==0),c3&&(k==0));
					res%=mod;
                }
			}
		}
	}
	return dp[p][r1][r2][r3][mx1][mx2][mx3][c1][c2][c3]=res;
}
int n;       
signed main(){
	ios::sync_with_stdio(0);
	cin.tie(0);
	cout.tie(0);
	for(int i=0;i<65;i++)
		for(int r1=0;r1<10;r1++)
			for(int r2=0;r2<10;r2++)
				for(int r3=0;r3<10;r3++)
					for(int mx1=0;mx1<2;mx1++)
						for(int mx2=0;mx2<2;mx2++)
							for(int mx3=0;mx3<2;mx3++) 
                            	for(int c1=0;c1<2;c1++)
                                	for(int c2=0;c2<2;c2++)
                                    	for(int c3=0;c3<2;c3++) dp[i][r1][r2][r3][mx1][mx2][mx3][c1][c2][c3]=-1;
	cin>>n;
	cin>>a1>>a2>>a3;
	while(n>0) dight.push_back(n%2),n/=2;
	cout<<solve(dight.size(),0,0,0,true,true,true,true,true,true)<<'\n';
	return 0;
}

2023/8/28 10:31
加载中...