数位dp。

一,思想:

在处理1e9甚至1e18,1e100的问题时,因为在统计情况下有很多重复的计算,数位dp实现了相同状态只计算一次,从而大幅减少运算时间,思想就是对每一位进行dp,计算时记忆化每一位可以有的状态。

如我们在统计1234的状态时,可以拆成统计0~10000,0~2000,0~300,0~40数位统计

我们用bit数组由低到高存储每一位,bit[1]=4,bit[2]=3,bit[3]=2,bit[4]=1.

然后dp从高位到低位进行

const int N = 20;
int dp[20][N],bit[N];
int dfs(int len,int sta,bool limit)//limit表示当前位有没有被bit限制
{
	if(!len)return (sta==1);//进行到最后一位,判断状态是否符合情况,符合就+1
	if(!limit&&dp[len][sta]!=-1)return dp[len][sta];//如果记忆化过,直接返回
	int ans=0;
	int top=limit?bit[len]:9;//如果有限制,如1234,当前len=4,那么top<=bit[4]=1,有限制,没有limit限制就可以0~9
	for(int i=0; i<=top; ++i)if(状态)ans+=dfs(len-1,newsta,limit&&i==top);//累积符合状态的情况
	if(!limit)return dp[len][sta]=ans;//如果没有限制,即我们现在计算出来符合该状态的所有情况数(没有被限制的)那么可以记忆下来
	return ans;
}

int cal(int x)//把数字x按位存储到数组
{
	int len=0;
	while(x)bit[++len]=x%10,x/=10;
	return dfs(len,0,1);
}

注意事项:

  1. 前导0问题,在数位dp中,0,000,0000是看成不同的,所以统计答案需要考虑他们是否有影响
  2. 继承问题,如从高位到低位计算0000123,前面4个0是初始状态,有时继承时需要特判考虑

例题1,Problem - 4507 (hdu.edu.cn)

思路:

  1. 对于数位dp求取与7无关的数,操作是简单的。无脑模拟即可
    1. 首先显然的,我们每一位没有7
    2. 然后他的状态就是前面各位的累积和模7,前面的数模7。如果当前len位时,前面的累积(dsum)模7相等且前面的数字(ssum)模7也相等,可以视作同一个状态
  2. 但是他加了个平方限制,即有继承关系,我们可以把每个平方数看成(A+B)^2.即我们对每个len可以记录他当前位符合条件的数的平方(ssum),与当前位符合条件的数的和(dsum)。
    1. 假设我们当前位的数字是i,如果我们已经记录了后代的B^2与B,后代一共有cnt个符合要求的数。
    2. 那么更新当前位ssum+=(cnt*i*10^{len})+(2*10^{len}*B)+B^{2}即(A+B)^2。dsum+=cnt*i*10^{len}+B,即a+b。
#include <bits/stdc++.h>
using namespace std;
#define ll               long long
#define endl             "\n"
#define int              long long
const int N = 21;
const int mod=1e9+7;
struct node
{
	int cnt,dsum,ssum;
	node()
	{
		cnt=-1,dsum=0,ssum=0;//记录B^2和与B和
	}
	node(int a,int b,int c):cnt(a),dsum(b),ssum(c) {};
} dp[N][N][N];
int bit[N],pre[N];

node dfs(int len,int dsum,int ssum,bool limit)
{
	if(!len)return (dsum&&ssum?node(1,0,0):node(-1,0,0));
	if(!limit&&dp[len][dsum][ssum].cnt!=-1)return dp[len][dsum][ssum];
	node ans;
	int top=limit?bit[len]:9;
	for(int i=0; i<=top; ++i)if(i!=7)
			{
				node tmp=dfs(len-1,(dsum+i)%7,(ssum*10+i)%7,limit&&i==top);
				if(tmp.cnt!=-1)//子代有符合情况的
					{
						if(ans.cnt==-1)ans.cnt=0;
						int A=i*pre[len]%mod;
						ans.cnt=(ans.cnt+tmp.cnt)%mod;
						ans.ssum=(ans.ssum+tmp.ssum+2*tmp.dsum*A%mod+tmp.cnt*A%mod*A%mod)%mod;
						ans.dsum=(ans.dsum+tmp.dsum+tmp.cnt*A%mod)%mod;
					}
			}
	if(!limit)return dp[len][dsum][ssum]=ans;
	return ans;
}

int cal(int x)
{
	int k=0;
	while(x)bit[++k]=x%10,x/=10;
	return dfs(k,0,0,1).ssum;
}
void mysolve()
{
	int l,r;
	cin>>l>>r;
	cout<<(cal(r)-cal(l-1)+mod)%mod<<endl;
}

int32_t main()
{
	std::ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
	ll t=1;
	cin >> t;
	pre[1]=1;
	for(int i=2; i<=20; ++i)pre[i]=pre[i-1]*10%mod;//预处理10^len
	while (t--)
		{
			mysolve();
		}
	system("pause");
	return 0;
}

例题2:Problem - 3709 (hdu.edu.cn)

思路:

  1. 数位只有18,而sum<=(9*19*18/2)<2000,我们可以枚举pos的每个位置来dp,如果最后sum=0,说明符合
  2. 如果sum<0,提前退出,不符合
  3. 所以他的状态是在当前枚举的是pos位,现在在len位时,累积sum的状态能获得的答案
  4. 需要考虑前导0,因为对于0,00,000000,哪个位置枚举pos都是一样的,重复计算len次,我们只需要1次。
#include <bits/stdc++.h>
using namespace std;
#define ll               long long
#define endl             "\n"
#define int              long long
const int N = 30;
int dp[N][N][2000],bit[N];

int dfs(int len,int sum,int pos,bool limit)
{
	if(!len)return sum==0;
	if(sum<0)return 0;
	if(!limit&&dp[len][pos][sum]!=-1)return dp[len][pos][sum];
	int ans=0;
	int top=limit?bit[len]:9;
	for(int i=0; i<=top; ++i)ans+=dfs(len-1,sum+i*(len-pos),pos,limit&&i==top);
	if(!limit)return dp[len][pos][sum]=ans;
	return ans;
}
int cal(int x)
{
	if(x<0)return 1;
	int k=0;
	while(x)bit[++k]=x%10,x/=10;
	int ans=0;
	for(int i=0; i<=k; ++i)ans+=dfs(k,0,i,1);//枚举pos
	return ans-k+1;//剔除多余前导0
}
void mysolve()
{
	int l,r;
	cin>>l>>r;
	cout<<cal(r)-cal(l-1)<<endl;
}

int32_t main()
{
	std::ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
	ll t=1;
	cin >> t;
	memset(dp,-1,sizeof(dp));
	while (t--)
		{
			mysolve();
		}
	system("pause");
	return 0;
}

例题3:Problem - 4352 (hdu.edu.cn)

思路:

  1. 首先,我们知道在O(n*logn)下就可以求出lis。
  2. 因为数位的特性,lis<=10,数字是0~9,我们可以用状压来表示他的当前状态。用数字x存储状态,如果x的二进制上i位为1,说明其lis含这个数字,我们更新时也是更新x
  3. 这样,数位的状态就是在求取lis为k时,当前长度为len时,其当前lic的状态为x可能的答案数。
#include <bits/stdc++.h>
using namespace std;
#define ll               long long
#define endl             "\n"
#define int              long long
const int N = 20;
int dp[N][2000][N],bit[N];
int k;
int tocnt(int x)//计数lis
{
	int cnt=0;
	for(int i=0; i<10; ++i)if(x&(1<<i))cnt++;
	return cnt;
}

int update(int x,int p)
{
	for(int i=p; i<10; ++i)if(x&(1<<i))return (x^(1<<i))|(1<<p);//更新lis就是贪心从比p第一个大的数更新,表示lis为相同长度时,用p可以更小
	return x|(1<<p);//找不到比p大的,那么lis长度加1,即加了个p
}

int dfs(int len,int x,bool limit)
{
	if(!len)return tocnt(x)==k;
	if(!limit&&dp[len][x][k]!=-1)return dp[len][x][k];
	int ans=0;
	int top=limit?bit[len]:9;
	for(int i=0; i<=top; ++i)ans+=dfs(len-1,(!x&&!i?0:update(x,i)),limit&&i==top);//!x&&!i?0:update(x,i)处理第一次继承问题,即前面都是0,如果当前i为0或为其他的情况
	if(!limit)return dp[len][x][k]=ans;
	return ans;
}

int cal(int x)
{
	int len=0;
	while(x)bit[++len]=x%10,x/=10;
	return dfs(len,0,1);
}

int tt=0;
void mysolve()
{
	int l,r;
	cin>>l>>r>>k;
	cout<<"Case #"<<++tt<<": ";
	cout<<cal(r)-cal(l-1)<<endl;
}

int32_t main()
{
	std::ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
	ll t=1;
	cin >> t;
	memset(dp,-1,sizeof(dp));
	while (t--)
		{
			mysolve();
		}
	system("pause");
	return 0;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值