洛谷：P3281 [SCOI2013]数数 （优秀的解法）

刷了这么久的数位 dp ，照样被这题虐，还从早上虐到晚上，对自己无语...（机房里又是只有我一个人，寂寞。）

题目：洛谷P3281 [SCOI2013]数数

题目描述

Fish 是一条生活在海里的鱼，有一天他很无聊，就开始数数玩。他数数玩的具体规则是：

1. 确定数数的进制B

2. 确定一个数数的区间[L, R]

3. 对于[L, R] 间的每一个数，把该数视为一个字符串，列出该字符串的每一个（连续的）子串对应的B进制数的值。

4. 对所有列出的数求和。现在Fish 数了一遍数，但是不确定自己的结果是否正确了。由于[L, R] 较大，他没有多余精力去验证是否正确，你能写一个程序来帮他验证吗？

输入输出格式

输入格式：

输入包含三行。

第一行仅有一个数B，表示数数的进制。

第二行有N +1 个数，第一个数为N，表示数L 在B 进制下的长度为N，接下里的N个数从高位到低位的表示数L 的具体每一位。

第三行有M+ 1 个数，第一个数为M，表示数R 在B 进制下的长度为M，接下里的M个数从高位到低位的表示数R 的具体每一位。

 

输出格式：

输出仅一行，即按照Fish 数数规则的结果，结果用10 进制表示，由于该数可能很大，输出该数模上20130427的模数。

分析：

数位 dp ，有点强大，又是一道需要感性理解的题目。。。

首先我们准备好 dp 状态： dp[i][2]  ，表示共有 i 位的 B 进制数，前缀子串的和。

dp[i][0] 表示无限制（甚至允许前导 0 的存在），

dp[i][1] 表示当前最高位不超过 d[i] （d[i] 表示读入的 B 进制数从后往前数的第 i 位）

那么 我们考虑 dp[i][0] 的转移：

假设当前 dp[i-1][0] 已经得到，那么 处理 dp[i][0] 就相当于 在 dp[i-1][0] 的基础上，在最高位（即第 i-1 位之前）加上一位 B 进制数 x 。

那么对于前缀子串的和来说，x 的贡献也就是让 后面的 i-1 个数字多了 x 中选择（0 ~ x-1），而 x 本身对于前缀子串和的贡献就是 x * B⁰ + x * B¹ +.....+ x*B^i-1 （看不懂可以多想几遍，注意抓住贡献这个关键）

那么 我们就可以列出转移式子了：$$ dp[i][0] = B × dp[i−1][0]+ \drac{B(B−1)}{2} × B[i] × S[i−1] (B[i] = B^{i} , S[i] = \sum_{1}^{i} B[i]) $$

而对于有限制的 dp[i][1] 我们也可以对应的得到式子： $$ dp[i][1] = d[i] × dp[i−1][0]+\drac{d[i](d[i]−1)}{2} × B[i-1] × S[i-1] + dp[i−1][1] + d[i]∗(sub[i−1]+1) × S[i-1] ( d 的意义同上，sub[i] 表示读入的 B 进制数的长度为 i 的后缀子串对应值) $$

那么 对于 ans 的累加就是： $$ ans=\sum_{i=1}^{len} max(0,pre[i+1]−1) × f[i][0]+f[i][1] $$ 也就是说我们一步一步去处理 dp 数组，然后就可以同时累加答案了

为什么 ans 这样累加？ 其实上面的式子就是在说，除去当前处理完的 i 位，剩下的（前缀子串对应值）有 pre[i+1] 种取法（因为可含前导零，所以方案数就是前缀子串对应数值）

但是前缀不能取到完全状态，于是 -1 。然后加上有限制的 dp[i][1] ，就是当前可累加的答案了 

emmmm...麻烦死了...其实不是很难理解，但是很难想到可以这么转移...之类的（都是借口）

于是...就可以上代码了吧？

代码

 //by Judge
 #include<iostream>
 #include<cstring>
 #include<cstdio>
 #define ll long long
 #define int long long
 using namespace std;
 const int M=1e5+;
 const ll mod=;
 //#define getchar() (p1==p2&&(p2=(p1=buf)+fread(buf,1,1<<21,stdin),p1==p2)?EOF:*p1++)
 char buf[<<],*p1=buf,*p2=buf;
 inline int read(){
     int x=,f=; char c=getchar();
     for(;!isdigit(c);c=getchar()) if(c=='-') f=-;
     for(;isdigit(c);c=getchar()) x=x*+c-''; return x*f;
 }
 ll B,len,ans,d[M],f[M]={},sum[M]={},pre[M],sub[M],dp[M][];
 inline void prep(){ //预处理
         for(int i=,j;i<=1e5;++i)
                 f[i]=f[i-]*B%mod,
                 sum[i]=(sum[i-]+f[i])%mod; }
 inline ll solv(){
     ll ans=pre[len+]=; //恶心的pre初始化，最后的坑
     for(int i=;i<=len;++i) sub[i]=(d[i]*f[i-]%mod+sub[i-])%mod;
     for(int i=len;i>=;--i) pre[i]=(pre[i+]*B%mod+d[i])%mod;
         //前缀值、后缀值的预处理
     for(int i=;i<=len;++i){ //dp 转移，查了半天发现没毛病
         dp[i][]=(dp[i-][]*B%mod+B*(B-)/%mod*f[i-]%mod*sum[i-]%mod)%mod;
         dp[i][]=(dp[i-][]*d[i]%mod+d[i]*(d[i]-)/%mod*f[i-]%mod*sum[i-]%mod)%mod;
         dp[i][]=(dp[i][]+dp[i-][]+d[i]*(sub[i-]+)%mod*sum[i-]%mod)%mod;
         ans=(ans+max(0ll,pre[i+]-)*dp[i][]%mod+dp[i][])%mod;
     } return ans;
 }
 signed main(){
     B=read(),len=read(),prep();
     for(int i=len;i;--i) d[i]=read();
     for(int i=;i<=len;++i) //繁杂的数字处理，坑
         if(d[i]){ --d[i]; break; }
         else d[i]=B-;
     if(!d[len]) --len;
     ans=-solv(), len=read();
     for(int i=len;i;--i) d[i]=read();
     ans+=solv(),printf("%lld\n",(ans%mod+mod)%mod); return ;
 }