洛谷P1831杠杆数（数位DP）

洛谷P1831杠杆数（数位DP）

题目传送门
今天第一次接触数位DP，还是有点费解的……
算法介绍：
数位DP，顾名思义就是跟数字有关。事实上，数位DP的题目类型基本是“求[l,r]中满足某某条件的数的个数”。通常，l，r的大小都会超过int甚至long long，因此O（n）的算法是行不通的。做这类题我们通常会使用“试填法”的思想，先利用DP预处理，再逐位枚举拼凑，或者直接使用记忆化搜索实现。俗话说数位DP就是另一种暴力枚举

---

思路：
再看这道题符合数位DP的特点，所以我们就用数位DP来做。
1：对于[x,y]这个区间中杠杆数的个数，我们只需要求[1,a]这个区间内的数，因为[x,y]区间内的数可以用[1,y]-[1,x-1]来实现（注意要把0特殊出来考虑）。 这个思想很关键，它保证了我们可以用数位DP来做，因为数位DP要枚举1到这个数之间的所有数。
2：然后我们对每个数枚举支点所在位置，对于每个符合条件的数，它的支点是唯一的。因为如果支点左移一位，支点左边的力矩和一定变小，右边力矩一定变大（试几组数就知道了），反之右移情况相同。
有了以上两点思路就可以解决问题了。

---

变量：
pos：表示这个数的位数。
num1：表示支点两边力矩和。当num1<0时，当前这个数不满足杠杆数，返回0：当num1==0时，不能马上返回1，而是要等到这一位枚举完再返回。
x：表示当前枚举到的位数。
lim：表示枚举时是否有限制。就比如说原来的数是114514，如果在4这一位枚举了3，那么5这一位就可以取0~9，如果4这一位枚举了4，那么为了保证枚举的这个数要小于114514，所以5这一位只能枚举0 ~ 5。
还有亿些细节看代码吧
AC代码：

#include<bits/stdc++.h>
#define int long long
using namespace std;
int l,r,dp[20][20][2005],num[205];
int dfs(int pos,int x,bool lim,int num1)
{
	if(!pos) return num1==0;
	if(!lim&&dp[pos][x][num1]!=-1) return dp[pos][x][num1];//小小记忆化
	int maxx=lim?num[pos]:9,ans=0;
	for(int i=0;i<=maxx;i++)
	{
		ans+=dfs(pos-1,x,lim&&i==maxx,num1+i*(pos-x));
	}
	if(!lim) dp[pos][x][num1]=ans;
	return ans;
}
int solve(int x)
{
	int ans=0,cnt=0;
	while(x)//拆分数位
	{
		num[++cnt]=x%10;
		x/=10;
	}
	for(int i=1;i<=cnt;i++)
	{
		ans+=dfs(cnt,i,1,0);
	}
	return ans-cnt+1;
}
signed main()
{
	memset(dp,-1,sizeof(dp));
	scanf("%lld%lld",&l,&r);
	if(!l) printf("%lld\n",solve(r));
	else printf("%lld\n",solve(r)-solve(l-1));
	return 0;
}

注意：solve（）函数最后返回的时候返回的是ans-cnt+1，而不是ans，因为如果枚举到00000……(cnt个0)，这个数会被算cnt遍，所以我们要减去cnt但是要保留一次，所以+1。