题目描述
数据范围为 L , R ≤ 1 0 500000 \mathtt{L,R \leq 10^{500000}} L,R≤10500000
题解与思路历程
这一看数据范围就能想到数位dp
然后数位dp都是套路差不多的方法。
考虑快速计算1 ~ n的答案,那么可以简单的用1 ~ R的答案减去1 ~ L-1的答案即可。
首先可以写出如下的框架:
void dp(int now,int lim,int sum) //now:当前dp到第几位,前一位是否有限制
//根据题目,还需要求出前面位中的逆序对。
{
if(!lim) { return count(len-now+1); } //没有限制则直接计算答案
if(now==len+1) { ans+=sum; return; } //一直到末位了,计算答案
{
//继续递归限制
}
//递归没有限制
}
考虑怎么求 c o u n t ( ) count() count()函数。
c o u n t ( ) count() count()函数需要做的就是求出剩下所有位随便填相互造成的"鲳对"数,以及对前面已经选择的数 造成的"鲳对"数
- 对于前者
考虑怎么求1 ~ 1 0 l e n 10^{len} 10len中的所有"鲳对":
若 l e n ≤ 1 len \leq 1 len≤1,则无"鲳对"
否则:
考虑第 i i i位摆 x x x,右边的 n − i n-i n−i位上任意的某一位若摆小于 x x x的数,则有一个"鲳对"
因为 x = 0...9 x=0...9 x=0...9,右边某一位的取值分别为 0...1 , 0...2 , . . . , 0...8 0...1,0...2,...,0...8 0...1,0...2,...,0...8
所以对于第
i
i
i位,右边某一位的取值方案有
45
45
45种,一共有
45
×
(
l
e
n
−
i
)
45 \times (len-i)
45×(len−i)种,每一位单独考虑
其余位置都有
1
0
l
e
n
−
2
10^{len-2}
10len−2的放数方法。
所以答案应该为
∑ i = 1 l e n 45 × ( l e n − i ) × 1 0 l e n − 2 \sum_{i=1}^{len}{45 \times (len-i) \times 10^{len-2}} i=1∑len45×(len−i)×10len−2
设上面这玩意儿为 s [ l e n ] s[len] s[len],显然可以 O ( n ) O(n) O(n)预处理出 s s s数组。
void init()
{
ten[0]=1;
long long x=45,t=0;
for(int i=1;i<=5e5;i++)
{
ten[i]=ten[i-1]*10LL%mod;
t+=i;
t%=mod;
s[i+1]=1ll*x*ten[i-1]%mod*t%mod;
}
}
- 对于后者
我们设 g s [ 0 gs[0 gs[0~ 9 ] 9] 9]表示前面已经选择了有哪些数(gs=个数),后面还剩 l e n len len个位置
对于 g s [ i ] gs[i] gs[i]显然有"鲳对" = g s [ i ] × 1 0 l e n − 1 × i × l e n =gs[i] \times 10^{len-1} \times i \times len =gs[i]×10len−1×i×len
所以得出 c o u n t ( ) count() count()函数为:
void count(int len)
{
ans+=s[len];
if(ans>mod) ans-=mod;
for(int i=1;i<10;i++)
ans=(ans+1ll*gs[i]*ten[len-1]%mod*(1ll*i*len)%mod)%mod;
}
- 但这样我们还漏掉了一个地方的贡献(就是这个地方导致我调了1h!!!)
那就是我们在dp时已知前面确定了的数字之间的"鲳对"为 s u m sum sum,那么后面我们不管怎么填都有至少 s u m sum sum个"鲳对"。
只要考虑好以上几点我们就可以很好的写出dp的所有部分了,最后记得是计算 L − 1 L-1 L−1的贡献
C o d e \mathcal{Code} Code
/*******************************
Author:galaxy yr
LANG:C++
Created Time:2019年10月09日 星期三 14时16分22秒
*******************************/
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<string>
#define int long long
using namespace std;
const int maxn=1e6+10;
const int mod=998244353;
int T,id,ten[maxn],s[maxn],gs[11],len;
long long ans,res;
string ss;
void init()
{
ten[0]=1;
long long x=45,t=0;
for(int i=1;i<=5e5;i++)
{
ten[i]=ten[i-1]*10LL%mod;
t+=i;
t%=mod;
s[i+1]=1ll*x*ten[i-1]%mod*t%mod;
}
}
void count(int len)
{
ans+=s[len];
if(ans>mod) ans-=mod;
for(int i=1;i<10;i++)
ans=(ans+1ll*gs[i]*ten[len-1]%mod*(1ll*i*len)%mod)%mod;
}
void dp(int now,int lim,int sum)
{
//没有限制,则计算答案
if(!lim) { ans=(ans+(sum*ten[len-now+1])%mod)%mod; return count(len-now+1) ;}
if(now==len+1) { ans+=sum; return; }
int c=ss[now]-'0',tmp=0;
//先考虑有限制
{
gs[c]++;
for(int i=c+1;i<10;i++)
{
tmp+=gs[i];
tmp%=mod;
}
dp(now+1,1,(sum+tmp)%mod);
gs[c]--;
}
//考虑没有限制
for(int i=0;i<c;i++)
{
gs[i]++;
tmp=0;
for(int j=i+1;j<10;j++)
tmp+=gs[j];
dp(now+1,0,(sum+tmp)%mod);
gs[i]--;
}
}
//计算L这个数的"鲳对"
void add()
{
for(int i=1;i<=len;i++)
{
int c=ss[i]-'0';
gs[c]++;
for(int j=c+1;j<10;j++)
ans-=gs[j];
ans=(ans+mod)%mod;
}
for(int i=0;i<10;i++)
gs[i]=0;
}
signed main()
{
//freopen("pair.in","r",stdin);
//freopen("pair.out","w",stdout);
ios::sync_with_stdio(false);
init();
cin>>T>>id;
while(T--)
{
ans=res=0;
cin>>ss;
len=ss.size(); ss="$"+ss;
dp(1,1,0); add();
res=ans,ans=0;
cin>>ss;
len=ss.size(); ss="$"+ss;
dp(1,1,0);
cout<<(ans-res+mod)%mod<<endl;
}
return 0;
}
一定要记得开long long
,考试时就是这个原因导致
W
A
WA
WA和
R
E
RE
RE