[USACO23FEB] Piling Papers G题解

洛谷[USACO23FEB] Piling Papers G

题目大意

给你一个长度为$n$的序列$a$，满足$1\le a_i\le 9$，以及两个整数$A,B$。有$q$次询问，每次询问给出$l,r$。对于每次询问，一开始有一个空串，对$a_l,a_{l+1},\dots ,a_r$依次操作，对于每一个$a_i$，你可以选择将$a_i$放在$x$的前面、后面、或者什么也不做。最后，将$x$中的数字依次相连得到一个整数，求能使这个整数在$[A,B]$之间的方案数的数量，输出对$10^9+7$取模。

$1\le n\le 300,1\le A,B\le 10^{18}$

题解

用差分将要求的部分变为$[0,B]$的方案数减去$[0,A-1]$的方案数，那么我们只需考虑求$[0,k]$的方案数。

设$f_{i,l,r,0/1/2}$表示前$i$个数形成的$r-l+1$个数字组成的数，和$k$从高到低的第$l$位到第$r$位组成的数的关系是小于、等于还是大于。转移就是从$f_{i,l+1,r,0/1/2}$和$f_{i,l,r-1,0/1/2}$转移到$f_{i,l,r,0/1/2}$。

设$k$的位数为$d$，那么对于序列$a$从$1$到$i$的答案$su{m}_i=f_{n,1,d,0}+f_{n,1,d,1}+\sum _{j=1}^{d-1}\sum _{p=0}^2{f}_{n,1,j,p}$。前两项求的是位数等于$d$且值小于等于$k$的方案数，最后一项求的是位数小于$d$的方案数。

设$ans{l}_{i,j}$表示当$k=A-1$时$a_i,a_{i+1}\dots ,a_j$的答案，则$ans{l}_{i,j}=au{m}_j-su{m}_{i-1}$。但实际上，这样算还麻烦了，我们其实并不需要求$su{m}_i$。对于同一个$i$，在枚举$j$时继续用先前的$f$数组，就能$O(n^2{\log }^{2}A)$预处理出$ans{l}_{i,j}$。
设$ans{r}_{i,j}$表示当$k=B$时$a_i,a_{i+1}\dots ,a_j$的答案，则同理可以$O(n^2{\log }^{2}B)$预处理出$ans{r}_{i,j}$。这样的话，查询就是$O(1)$的了。

总时间复杂度为$O(n^2{\log }^{2}B+q)$。

#include<bits/stdc++.h>
using namespace std;
const long long mod=1000000007;
int n,q,v1,a[305],v[25];
long long ansl[305][305],ansr[305][305],f[25][25][3];
long long L,R;
void dd(long long x){
	v1=0;
	while(x){
		v[++v1]=x%10;x/=10;
	}
	reverse(v+1,v+v1+1);
}
int gt(int x,int y){
	if(x<y) return 0;
	else if(x==y) return 1;
	else return 2;
}
int main()
{
	scanf("%d%lld%lld",&n,&L,&R);
	for(int i=1;i<=n;i++){
		scanf("%d",&a[i]);
	}
	dd(L-1);
	for(int i=1;i<=n;i++){
		memset(f,0,sizeof(f));
		for(int j=i;j<=n;j++){
			for(int l=1;l<=v1;l++){
				for(int r=v1;r>l;r--){
					if(a[j]>v[l]){
						for(int k=0;k<=2;k++)
						f[l][r][2]=(f[l][r][2]+f[l+1][r][k])%mod;
					}
					else if(a[j]==v[l]){
						for(int k=0;k<=2;k++)
						f[l][r][k]=(f[l][r][k]+f[l+1][r][k])%mod;
					}
					else{
						for(int k=0;k<=2;k++)
						f[l][r][0]=(f[l][r][0]+f[l+1][r][k])%mod;
					}
					f[l][r][2]=(f[l][r][2]+f[l][r-1][2])%mod;
					f[l][r][gt(a[j],v[r])]=(f[l][r][gt(a[j],v[r])]+f[l][r-1][1])%mod;
					f[l][r][0]=(f[l][r][0]+f[l][r-1][0])%mod;
				}
			}
			for(int p=1;p<=v1;p++) f[p][p][gt(a[j],v[p])]=(f[p][p][gt(a[j],v[p])]+2)%mod;
			ansl[i][j]=(ansl[i][j]+f[1][v1][0]+f[1][v1][1])%mod;
			for(int p=1;p<v1;p++){
				for(int k=0;k<=2;k++)
				ansl[i][j]=(ansl[i][j]+f[1][p][k])%mod;
			}
		}
	}
	dd(R);
	for(int i=1;i<=n;i++){
		memset(f,0,sizeof(f));
		for(int j=i;j<=n;j++){
			for(int l=1;l<=v1;l++){
				for(int r=v1;r>l;r--){
					if(a[j]>v[l]){
						for(int k=0;k<=2;k++)
						f[l][r][2]=(f[l][r][2]+f[l+1][r][k])%mod;
					}
					else if(a[j]==v[l]){
						for(int k=0;k<=2;k++)
						f[l][r][k]=(f[l][r][k]+f[l+1][r][k])%mod;
					}
					else{
						for(int k=0;k<=2;k++)
						f[l][r][0]=(f[l][r][0]+f[l+1][r][k])%mod;
					}
					f[l][r][2]=(f[l][r][2]+f[l][r-1][2])%mod;
					f[l][r][gt(a[j],v[r])]=(f[l][r][gt(a[j],v[r])]+f[l][r-1][1])%mod;
					f[l][r][0]=(f[l][r][0]+f[l][r-1][0])%mod;
				}
			}
			for(int p=1;p<=v1;p++) f[p][p][gt(a[j],v[p])]=(f[p][p][gt(a[j],v[p])]+2)%mod;
			ansr[i][j]=(ansr[i][j]+f[1][v1][0]+f[1][v1][1])%mod;
			for(int p=1;p<v1;p++){
				for(int k=0;k<=2;k++)
				ansr[i][j]=(ansr[i][j]+f[1][p][k])%mod;
			}
		}
	}
	scanf("%d",&q);
	while(q--){
		int l,r;
		scanf("%d%d",&l,&r);
		printf("%lld\n",(ansr[l][r]-ansl[l][r]+mod)%mod);
	}
	return 0;
}