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

转载于 2018-10-15 08:44:00 发布 · 67 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

原文链接：http://www.cnblogs.com/zxyoi/p/10047222.html

本文深入探讨数位DP的非递归写法，通过结构体重载运算符优化算法，详细解析状态转移方程，分享从最高位开始的统计策略，并提供完整代码实现。

BZOJ传送门

洛谷传送门

解析：

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

思路：

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

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

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

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

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

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

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

怎么统计？

我们考虑从最高位开始，比如一个数可以分解成 $a*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;
}