Poj 1790 Base Numbers【题解报告|DP】

题目链接

题目大意

给你一个字符串，里面的前一部分是一个数，后一部分是表示他的进制，问这个字符串可以表示成多少个数。

思路分析

我们遍历每一个可能的进制组合，比如$1234$，遍历$4,34,234$，对每一个进制，用$cnt[i]$表示前$i$位能表示的合法数字的个数，这道题满足最优子结构，即如果假设$cnt[i-1]...cnt[0]$已经得到了，那么能够从他们中得到$cnt[i]$

假设当前进制$t$是一个$r$位的数字，那么我们要求输入字符的前$n-r$位能够凑成多少合法数字，假设$cnt[i-1],cnt[i-2],...,cnt[0]$已经得到了，那么如果我们固定最后一个数字(该数字的位数最多为$r$位)，

• 我们取$s[0:i]$的后$r$位作为一个数字，即$s[i-r:i]$是一个合法的数字且小于$t$，那么$cnt[i]+=cnt[i-r]$
• 如果我们取后$r-1$位作为一个数字，显然他比$t$小（因为位数少），因此$cnt[i]+=cnt[i-(r-1)]$
• …
• 直到我们取最后一位数字作为一个数，有$cnt[i]+=cnt[i-1]$

需要注意的是，在这个过程中，我们需要保证取得的数字没有前导0，剩下一切尽在注释中

#define ll long long
#define vec vector<int>
#define inf 0x3f3f3f3f
#define P pair<int,int>
#define MAX 40

string s;
int cnt[MAX], len;

//后r位为进制位能得到的数字有多少个
ll cnt_num(int r) {
	string t = s.substr(len - r);//t为当前的进制
	if (t[0] == '0')return 0; //进制的首位不能为0
	if (s[0] == '0'&&len - r > 1)return 0;//数字的首位为0，考虑0123，只有一种方式：（0）123

	//遍历除了进制外的前len-r位数,cnt[i]表示前i位能表示的合法数字的个数
	memset(cnt, 0, sizeof(cnt));
	//初始化第一位
	if (r > 1 || (r == 1 && s[0] < t[0]))cnt[0] = 1;
	else return 0;//最后一位作为进制位，数的第一位已经大于他了。

	for (int i = 0; i <= len - r; i++) {//前i位能凑出多少个合法数字
		/*
			假设cnt[i-1]已经得到了，那么应该如何求cnt[i]
			当前进制t是一个r位的数字，那么如果 
			我们取s[0:i]的后r位作为一个数字，即s[i-r:i]是一个合法的数字且小于t，那么cnt[i]+=cnt[i-r]
			如果我们取后r-1位作为一个数字，显然他比t小（因为位数少），因此cnt[i]+=cnt[i-(r-1)]
			...
			直到我们取最后一位数字作为一个数，有cnt[i]+=cnt[i-1]
		*/
		int beg = i - r;
		if (beg < 0)beg = 0;
		for (int j = beg; j < i; j++) {
			//j != i - 1 && s[j] == '0':数字超过一位，而且有前导0
			if (j != i - 1 && s[j] == '0')continue;
			if (j + r == i && s.substr(j, r) >= t)continue;//位数相等且数字不小于进制数
			//s[0:j]-s[j:i]能够是一个合法的数字
			cnt[i] += cnt[j];
		}
	}
	return cnt[len - r];
}

int main() {
	while (cin >> s && s != "#") {
		ll res = 0; 
		len = s.size();

		for (int i = 1; i < s.size(); i++) 
			res += cnt_num(i);

		if (res > 0)printf("The code %s can represent %lld numbers.\n", s.c_str(), res);
		else printf("The code %s is invalid.\n", s.c_str());
	}
}