标签：剪枝 cnt flagb flaga 748 Codeforces pos int maxn

F. Red-Black Number
题意:给你一个n位的数字 让你把n位数分别染上红色或者黑色 染上红色的需要整除A个数记为r,染上黑色需要整除B个数记为b,让我们求abs(r - b)最小,2 <= n <= 40, 1 <= A,B <= 40
思路:首先我们可以看到,有n位数字分成两半,那么我们一定可以想到暴力枚举所有方案就有2^n - 1种方案对吧,那么之后我们就可以想如何进行优化.

对于染成红色的数之和记作flaga, 染成红色的数记作flagb, 那么我们可以看到对于整除A,B来说,我们可以考虑模数,那么就只有A,B种状态,最多就只有40 * 40种,那么之后我们就可以利用多维dp数组进行优化.

第一维度 当前记忆化搜索到第几位数字
第二维度 当前整除A后的模数flaga
第三维度 当前整除B后的模数flagb
第四维度 当前染成红色的个数r,这样就做完了暴力搜索,然后通过数组进行剪枝进行复杂度优化就好了。

#include<bits/stdc++.h>
#define IOS ios::sync_with_stdio(false); cin.tie(0); cout.tie(0)
#define ll long long
using namespace std;
int n,a,b;
int ans;
const int maxn = 45;
int dp[maxn][maxn][maxn][maxn];
int col[maxn];
int vis[maxn];
string str;
void dfs(int pos, int flaga, int flagb, int cnt){
	if(pos == n){
		if(flaga == 0 && flagb == 0 && cnt > 0 && cnt < n){
			if(abs(n - 2*cnt) < ans){
				ans = abs(n - 2*cnt);
				for(int i = 0; i < n; i++){
					col[i] = vis[i];
				}
			}
		}
		return ;
	}
	if(dp[pos][flaga][flagb][cnt])return ;
	dp[pos][flaga][flagb][cnt] = 1;
	vis[pos] = 0;
	dfs(pos + 1, (flaga*10 + str[pos] - '0')%a, flagb, cnt + 1); 
	vis[pos] = 1;
	dfs(pos + 1, (flaga), (flagb*10 + str[pos] - '0')%b, cnt);
	return ;
}
void solve(){
	//给你一个n位的数字 让你把n位数分别染上红色或者黑色 
	//染上红色的需要整除A个数记为r  染上黑色需要整除B个数记为b
	//让我们求abs(r - b)最小
	cin >> n >> a >> b;
	cin >> str;
	memset(dp, 0, sizeof dp);
	ans = 1e9;
	dfs(0, 0, 0, 0);
	if(ans == 1e9){
		cout << "-1" << '\n';
	}else{
		for(int i = 0; i < n; i++){
			if(col[i] == 0)cout << 'R';
			else cout << 'B';
		}
		cout << '\n';
	}
}
int main(){
	//freopen("2.in","r",stdin);
	IOS;
	int t;
	cin >> t;
	while(t--)solve();
}

标签：剪枝,cnt,flagb,flaga,748,Codeforces,pos,int,maxn
来源： https://blog.csdn.net/weixin_45908221/article/details/120809950

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。