题目给了个f(x)的定义:F(x) = An * 2n-1 + An-1 * 2n-2 + ... + A2 * 2 + A1 * 1,Ai是十进制数位,然后给出a,b求区间[0,b]内满足f(i)<=f(a)的i的个数。
常规想:这个f(x)计算就和数位计算是一样的,就是加了权值,所以dp[pos][sum],这状态是基本的。a是题目给定的,f(a)是变化的不过f(a)最大好像是4600的样子。如果要memset优化就要加一维存f(a)的不同取值,那就是dp[10][4600][4600],这显然不合法。
这个时候就要用减法了,dp[pos][sum],sum不是存当前枚举的数的前缀和(加权的),而是枚举到当前pos位,后面还需要凑sum的权值和的个数,
也就是说初始的是时候sum是f(a),枚举一位就减去这一位在计算f(i)的权值,显然sum》
=0时就是满足的,后面的位数凑足sum位就可以了。
#include <bits/stdc++.h>
using namespace std;
const int maxn = 12;
const int maxm = 10010;
int dp[maxn][maxm];
int a[maxn];
int pow2[maxn];
int len = 0;
void init(){
pow2[0] = 1;
for(int i = 1;i < maxn;i++)
pow2[i] = pow2[i - 1] * 2;
memset(dp,-1,sizeof(dp));
}
int b[maxn];
int calc(int x){
int ans = 0, pos = 0;
memset(b, 0, sizeof(b));
while(x > 0){
b[pos++] = x % 10;
x /= 10;
}
while(pos > 0){
pos--;
ans = ans * 2 + b[pos];
}
return ans;
}
int dfs(int pos, int state, int lead, int limit){
int ans = 0;
if(pos < 0) {
if(state >= 0) return 1;
else return 0;
}//printf("pos: %d\n",pos);
if(state < 0) return 0;
// printf("state: %d\n",state);
if(!limit && dp[pos][state] != -1) return dp[pos][state];
// printf("pos2: %d\n",pos);
int up = limit ? a[pos] : 9;
// printf("a[pos]: %d\n",a[pos]);
for(int i = 0;i <= up;i++){
int temp = i * pow2[pos];
// if(state - temp < 0) continue;
ans += dfs(pos - 1, state - i * pow2[pos], lead && (i == 0), limit && (i == up));
}
if(!limit && dp[pos][state] == -1) dp[pos][state] = ans;
// printf("ans: %d\n",ans);
return ans;
}
int solve(int aa, int bb){
int temp = calc(aa);
// printf("temp: %d\n",temp);
int pos = 0,len = 0;
while(bb > 0){
a[pos++] = bb % 10;
bb/=10;
len++;
}
return dfs(pos - 1, temp ,true, true);
}
int main()
{
int kase;scanf("%d",&kase);
int T = 0;
init();
int aa, bb;
while(kase--){
scanf("%d%d",&aa,&bb);
int temp = calc(aa);
printf("Case #%d: %d\n",++T,solve(aa, bb));
}
return 0;
}
=0时就是满足的,后面的位数凑足sum位就可以了。
仔细想想这个状态是与f(a)无关的(新手似乎很难理解),一个状态只有在sum>=0时才满足,如果我们按常规的思想求f(i)的话,那么最后sum=f(a)才是满足的条件。