Codeforces Round #748 (Div. 3) F - Red-Black Number （记忆化搜索）

最新推荐文章于 2024-09-20 11:31:31 发布

豆沙睡不醒

最新推荐文章于 2024-09-20 11:31:31 发布

阅读量185

点赞数 1

分类专栏： # ACM-题目合集 # ACM-计算几何/思维/构造文章标签：算法动态规划 html

本文链接：https://blog.csdn.net/weixin_45675097/article/details/120772604

版权

ACM-题目合集同时被 2 个专栏收录

106 篇文章 5 订阅

订阅专栏

ACM-计算几何/思维/构造

94 篇文章 3 订阅

订阅专栏

linkkk

题意

给出长度为 $n (n < = 40)$ 由数字构成的字符串和 $a, b$ ，请将该字符串分为两部分 $A, B$ ，使得 $A$ 变为数字后可以整除 $a$ ， $B$ 可以整除 $b$ ，并且 $A, B$ 至少有一位。最小化 $A, B$ 的位数之差，输出方案。

思路

看到 $n$ 的范围后想到了搜索，会有很多重复的状态所以做一个标记，如果已经遇到过这个状态就不再继续向下搜了。
搜索的时候维护当前 $A\mod a$ 的值 $m a$ ， $\mod b$ 的值 $m b$ ， $A$ 的位数 $c n t a$ 和当前枚举到第 $u$ 位。终止条件为 $u = = n$ ，看此时能否满足整除条件并且 $1 < = c n t a < n$ 。

代码

#include<iostream>
#include<cstdio> 
#include<string>
#include<ctime>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<stack>
#include<queue>
#include<map>
#include<set>
#include<sstream>
using namespace std;


typedef long long ll;

int n, a, b, ans, col[55], dp[55][55][55][55];
string s;
int st[55];

void dfs(int u, int ma, int mb, int cnta) {
	if (u == n) {
		if (!ma && !mb && cnta >= 1 && cnta < n) {
			if (abs(n - cnta - cnta) < ans) {
				ans = abs(n - cnta - cnta);
				for (int i = 0; i < n; i++) col[i] = st[i];
			}
		}
		return;
	}
	if (dp[u][ma][mb][cnta]) return;
	dp[u][ma][mb][cnta] = 1;
	st[u] = 0;
	dfs(u + 1, (ma*10 + s[u] - '0') % a, mb, cnta + 1);
	st[u] = 1;
	dfs(u + 1, ma, (mb*10 + s[u] - '0') % b, cnta);
}

int main() {
	int _; cin >> _;
	while (_--) {
		cin >> n >> a >> b;
		cin >> s;
		ans = 1e9;
		memset(dp, 0, sizeof dp);
		dfs(0, 0, 0, 0);
		if (ans == 1e9) puts("-1");
		else {
			for (int i = 0; i < s.size(); i++)
				if (col[i]) cout << "B";
				else cout << "R";
			puts("");
		}
	}




	return 0;
}
/*
4
5 3 13
02165
4 2 1
1357
8 1 1
12345678
2 7 9
90

*/