[CF908G]New Year and Original Order

题目

传送门 to luogu

思路

数位 $\mathtt{d}\mathtt{p}$ 难倒英雄汉 😭

我们有两类思路：直接填数字，计算对应的 $f$ 之和；对于每个 $f$，计算其原数的数量，然后统计权值。哪一个更好呢？

事实上都不好。第一个非常难进行，因为 $f$ 的变化很大；第二个的复杂度较高，因为方案数不容易计算。可以做到 $\mathcal{O}(10^2{n}^3)$ 是真的。如果你推生成函数，也可以推出 $\mathcal{O}(10n\log n)$ 的做法。$\mathrm{l}\mathrm{l}\mathrm{s}\mathrm{w}orzorzorz$！

所以这里有一个神来之笔：每种数字分开算贡献。然后第一种思路就可以做了。很简单的道理：如果下一个数字比 $d$ 大，那么所有已有的 $d$ 在 $f$ 中都会前进一位，也就是 $\times 10$；如果小，则没有任何影响。如果恰好是 $d$ 呢？

再引入一个 $g(i)$ 表示，对于所有已经有了 $i$ 个数位的数字，如果下一个数字是 $d$，其在 $f$ 中的权值（即 $10^{?}\times d$ ）的和。与 $f$ 类似，可以直接递推，即
$$g(i)=d\cdot g(i-1)+(9-d)\cdot 10\cdot g(i-1)+g(i-1)$$

我们认为新加入的 $d$ 在所有已有的 $d$ 的最后，所以这一位是 $d$ 不会让 $g$ 的值变化。然后你可以写出
$$\begin{gathered} f(i)=d\cdot f(i-1)+(9-d)\cdot 10\cdot f(i-1) \\ +[10\cdot f(i-1)+d\cdot g(i-1)] \end{gathered}$$

由于有点长，我打了个换行。初值是 $g(0)=1$，即没有数字。

具体的代码，再加上一个 $0/1$ 表示是否顶住上界就行。复杂度 $\mathcal{O}(Bn)$（这里 $B=10$ 表示进制）。

代码

由于分类讨论比较麻烦，我们可以舍弃常数的优秀。记住：如果不是乱搞骗分，正确性比复杂度更重要。

#include <cstdio>
#include <iostream>
#include <cstring>
using namespace std;
typedef long long int_;
inline int readint(){
	int a = 0; char c = getchar(), f = 1;
	for(; c<'0'||c>'9'; c=getchar())
		if(c == '-') f = -f;
	for(; '0'<=c&&c<='9'; c=getchar())
		a = (a<<3)+(a<<1)+(c^48);
	return a*f;
}

const int Mod = 1e9+7;
const int MaxN = 705;
int f[MaxN][2], g[MaxN][2];
int num[MaxN], n;
int solve(int d){
	memset(g,0,(n+1)*2<<2);
	memset(f,0,(n+1)*2<<2);
	g[0][0] = 1; int_ v;
	for(int i=1; i<=n; ++i)
	for(int j=0; j<10; ++j)
	for(int p=j>num[i]; p<2; ++p){
		v = g[i-1][p]*((j>d)?10ll:1);
		g[i][p|(j<num[i])] = (v+
			g[i][p|(j<num[i])])%Mod;
		
		v = f[i-1][p]*((j>=d)?10ll:1);
		if(j == d) v += 1ll*g[i-1][p]*d;
		f[i][p|(j<num[i])] = (v+
			f[i][p|(j<num[i])])%Mod;
	}
	return f[n][1];
}

char str[MaxN];
int main(){
	scanf("%s",str+1);
	n = strlen(str+1);
	for(int i=1; i<=n; ++i)
		num[i] = str[i]-'0';
	++ num[n]; int ans = 0;
	for(int i=1; i<=9; ++i)
		ans = (ans+solve(i))%Mod;
	printf("%d\n",ans);
	return 0;
}

不过嘛，精益求精，我也写了一个真正 $\mathcal{O}(Bn)$ 的，仅供参考。也确实跑得快。

#include <cstdio>
#include <iostream>
#include <cstring>
using namespace std;
typedef long long int_;
inline int readint(){
	int a = 0; char c = getchar(), f = 1;
	for(; c<'0'||c>'9'; c=getchar())
		if(c == '-') f = -f;
	for(; '0'<=c&&c<='9'; c=getchar())
		a = (a<<3)+(a<<1)+(c^48);
	return a*f;
}

const int Mod = 1e9+7;
const int MaxN = 705;
int num[MaxN], n;
int solve(int d){
	int_ g0 = 1, f0 = 0, g1 = 0, f1 = 0;
	for(int i=1,t; i<=n; ++i){
		/* one to one */ ;
		f1 = ((100-9*d)*f1+d*g1)%Mod;
		g1 = (91-9*d)*g1%Mod;
		/* zero to one */ ;
		f1 += (t = min(num[i],d))*f0%Mod;
		if(d < num[i]) f1 += d*g0%Mod;
		f1 += (num[i]-t)*10*f0%Mod;
		g1 += (t = min(num[i],d+1))*g0%Mod;
		g1 += (num[i]-t)*10*g0%Mod;
		/* zero to zero */ ;
		f0 = ( f0*((num[i]<d)?1:10)+
			(num[i]==d)*d*g0 )%Mod;
		g0 = g0*((num[i]>d)?10:1)%Mod;
	}
	return f1%Mod;
}

char str[MaxN];
int main(){
	scanf("%s",str+1);
	n = strlen(str+1);
	for(int i=1; i<=n; ++i)
		num[i] = str[i]-'0';
	++ num[n]; int ans = 0;
	for(int i=1; i<=9; ++i)
		ans = (ans+solve(i))%Mod;
	printf("%d\n",ans);
	return 0;
}