HDU 5106 Bit's Problem 数位DP(求和)

题意;给出n,定义数x为合法,当x的二进制表示中,bit为1的位数有n位.
n<=1000,R<=2^1000.(给出的R用二进制表示)问[0,R)合法的数之和?


数位DP容易求[0,R)合法数的个数,怎么求合法数之和? 当第i位为1时 算从该状态从发能得到的合法个数*该位置权重.
注意点dp必须要二元组 否则标记后直接返回个数 不会计算位数的权值.

dp[pos][cnt] 状态[第pos位,需要bit个1]时合法的个数 和 从第pos位出发能得到的贡献值

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=1e3+20,M=3e2+5,mod=1e9+7; 
struct node{
	ll num,sum;
	node(ll a,ll b)
	{
		num=a,sum=b;
	}
	node(){}	
}dp[N][N];
ll d[N];
char c[N];
int n,digit[N];
void init()
{
	for(int i=0;i<N;i++)
		for(int j=0;j<N;j++)
			dp[i][j].sum=-1,dp[i][j].num=0;
	d[0]=1;
	for(int i=1;i<N;i++)
		d[i]=d[i-1]*2ll%mod;
}
node dfs(int pos,int limit,int bit)
{
	if(bit<0)	return node(0,0);
	if(pos==-1)
	{
		if(!bit)
			return node(1,0);
		return node(0,0);
	}
	if(!limit&&dp[pos][bit].sum!=-1)
		return dp[pos][bit];
	node ans=node(0,0);
	int up=limit?digit[pos]:1;
	for(int i=0;i<=up;i++)
	{
		node tmp=dfs(pos-1,limit&&i==up,bit-i);
		ans.num=(ans.num+tmp.num)%mod;
		ans.sum=(ans.sum+tmp.num*i*d[pos]%mod+tmp.sum)%mod;
	}
	if(!limit)
		dp[pos][bit]=ans;
	return ans;
}
void solve()
{
	int len=strlen(c);
	for(int i=0;c[i];i++)
		digit[i]=c[len-i-1]-'0';
	//R-1
	for(int i=0;i<len;i++)
		if(digit[i]==1)
		{
			digit[i]=0;
			break;
		}
		else
			digit[i]=1;
	node ans=dfs(len-1,1,n);
	printf("%lld\n",ans.sum);
}
int main()
{
	init();
	while(~scanf("%d%s",&n,c))
		solve();	
	return 0;
}