2018.10.15【BZOJ1833】【洛谷P2602】【ZJOI2010】数字计数（数位DP）

BZOJ传送门

洛谷传送门

---

解析：

感觉不是一个很好做的数位DP。

思路：

我其实是看了$hzwer$的题解才会这道题的非递归写法，结构体重载运算符大法好啊$orz$。

定义数组$f[i][j][k]$表示位数为$i$，最高位为$j$，有数字$k$的个数。

其中最后一维$k$可以封装一个结构体，因为我们接下来要对它做很多次的整体加法。

我们先预处理出所有的$f$数组，考虑$DP$处理。

显然有如下状态转移方程$$f[i][k]+=f[i][j]$$
这是整体加法，以及$$f[i][k][k]+=10^i$$

两步都很显然，第一步是统计除开最高位的低位，第二步就是统计最高位的贡献。

之后就是数位$DP$老套路了，分开统计$1-(l-1)$和$1-r$的答案，直接减法就是最终的答案。

怎么统计？

我们考虑从最高位开始，比如一个数可以分解成$a\ast 10^k+b$，其中$a$是一个在$[1,9]$的整数，b是一个不大于$10^k$的数，那么仅考虑这里的贡献，答案中包含的贡献就有$f[k][a]$以及在$a$的出现次数中还有一个$b+1$次的贡献。

然后按照上面的思路递归处理。
当然，这个递归很轻易就写成了非递归。

---

代码：

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define re register
#define gc getchar
#define pc putchar
#define cs const

inline ll getint(){
	re ll num;
	re char c;
	while(!isdigit(c=gc()));num=c^48;
	while(isdigit(c=gc()))num=(num<<1)+(num<<3)+(c^48);
	return num;
} 

struct data{
	ll a[10];
	data(){memset(a,0,sizeof a);}
	ll &operator[](cs int &offset){return a[offset];}
	friend data operator+(cs data &a,cs data &b){
		data res;
		for(int re i=0;i<10;++i)res[i]=a.a[i]+b.a[i];
		return res;
	}
}f[25][10];

ll l,r,t[20];

data solve(ll x){
	data ans;
	if(x==0)return ans[0]=1,ans;
	int len=13;
	while(t[len]>x)--len;
	for(int re i=1;i<len;++i)
	for(int re j=1;j<10;++j)
	ans=ans+f[i][j];
	++ans[0];
	ll now=x/t[len];
	for(int re i=1;i<now;++i)ans=ans+f[len][i];
	x%=t[len];
	ans[now]+=x+1;
	for(int i=len-1;i;--i){
		now=x/t[i];
		for(int re j=0;j<now;++j)
		ans=ans+f[i][j];
		x%=t[i];
		ans[now]+=x+1;
	}
	return ans;
}

signed main(){
	t[1]=1;
	for(int re i=2;i<=13;++i)t[i]=t[i-1]*10;
	for(int re i=0;i<10;++i)f[1][i][i]=1;
	for(int re i=2;i<=12;++i)
	for(int re k=0;k<10;++k)
	for(int re j=0;j<10;++j){
		f[i][k]=f[i][k]+f[i-1][j];
		f[i][k][k]+=t[i-1];
	}
	l=getint();
	r=getint();
	data res1=solve(l-1);
	data res2=solve(r);
	for(int re i=0;i<10;++i)cout<<res2[i]-res1[i]<<" ";
	return 0;
}