LightOJ-1032-数位dp

题目大意：给定n，fun(n)为n的二进制中有多少对相邻的1，求0到n所有的f(n)之和；

题目解析：很明显的一道数位dp；

定义 dp[i][0]为长度为i并且最高位为0的二进制数出现相邻1的总次数；

dp[i][1]为长度为i并且最高位为1的二进制数出现相邻1的总次数；

预处理：

dp[i][0]=dp[i-1][0]+dp[i-1][1];
dp[i][1]=dp[i-1][1]+dp[i-1][0]+(1<<(i-2));

(在前两个的基础上+0，(1<<(i-2))是最高位和其次都是1，那么剩下的随便取)

那么我们开始从高位到低位模拟，定义cnt为在当前位置，高位已经出现过cnt对相邻的1，如果cnt[i]=1，那么ans+=dp[i][0](把这位变成0),并且ans+=cnt*(1<<(i-1))，（后面的可以随便取，因为上面已经有相邻的1）；

AC代码：

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<string>
using namespace std;
typedef long long ll;
ll dp[35][2];
void init()
{
	int i;
	memset(dp,0,sizeof(dp));
	dp[2][1]=1;
	for(i=3;i<=35;i++)
	{
		dp[i][0]=dp[i-1][0]+dp[i-1][1];
		dp[i][1]=dp[i-1][1]+dp[i-1][0]+(1<<(i-2));
	}		
}
ll fun(ll n)
{
	
	ll ans,cnt,i,bit[35],len=0;
	while(n)
	{
		bit[++len]=n%2;
		n/=2;
	}
	bit[len+1]=0;
	ans=0;
	cnt=0;
	for(i=len;i;i--)
	{	
		if(bit[i]==1)
		{
			ans+=(ll)dp[i][0]; 
			ans+=(ll)cnt*(1<<(i-1));
		}		
		if(bit[i]==1&&bit[i+1]==1)
		{
			cnt++;	
		}	
	}
	ans+=cnt; 
	return ans;
}
int main()
{
	int c,cas;
	ll n;
	init();
	scanf("%d",&cas);
	for(c=1;c<=cas;c++)
	{
		scanf("%lld",&n);
		printf("Case %d: %lld\n",c,fun(n));
	}
	return 0;
}