codeforces1110H Modest Substrings

題面

題意

給出三個數$l,r,n$，求出一個由數字組成的字符串，使其所有子串中，符合在$[l,r]$內的數量最多，若有多組解，輸出字典序最小的一個。

做法

首先考慮最樸素的做法，可以將[l,r]內的所有數字當作字符串加入一個AC自動機，直接在上面dp。
但是因爲l和r極大，因此需要優化。發現當一個數的長度確定，並且某一個前綴也確定之後，無論其後面填什麼數，它都在$[l,r]$內，因此可以對AC自動機上的每一個節點多維護一個數組sum[n]，sum[i]表示當前狀態開始無論後面i個數是什麼，都可以對答案作出sum[i]的貢獻，然後將l,r加入自動機即可進行dp。

代碼

#include<bits/stdc++.h>
#define L 810
#define N 2010
#define M 20010
using namespace std;

int n,tt,l1,l2,dp[N][M],to[N][M];
char a[L],b[L];
struct Node
{
    int to[10],sum[N],fail;
}node[M];
queue<int>que;

inline int get(int u,int v)
{
    if(!node[u].to[v]) node[u].to[v]=++tt;
    return node[u].to[v];
}

inline void in()
{
    int i,j,k,u,v,p,q,t;
    u=v=1;
    if(l1==l2)
    {
	for(i=1;i<=l1;i++)
	{
	    p=a[i]-'0',q=b[i]-'0';
	    if(u==v)
	    {
		for(j=p+1;j<q;j++) node[get(u,j)].sum[l1-i]++;
		u=get(u,p),v=get(v,q);
	    }
	    else
	    {
		for(j=p+1;j<=9;j++) node[get(u,j)].sum[l1-i]++;
		for(j=0;j<q;j++) node[get(v,j)].sum[l1-i]++;
		u=get(u,p),v=get(v,q);
	    }
	}
	node[u].sum[0]++;
	if(u!=v) node[v].sum[0]++;
    }
    else
    {
	for(i=1;i<=l1;i++)
	{
	    p=a[i]-'0';
	    for(j=p+1;j<=9;j++) node[get(u,j)].sum[l1-i]++;
	    u=get(u,p);
	}
	node[u].sum[0]++;

	for(i=1;i<=l2;i++)
	{
	    q=b[i]-'0';
	    for(j=(i==1);j<q;j++) node[get(v,j)].sum[l2-i]++;
	    v=get(v,q);
	}
	node[v].sum[0]++;

	for(i=1;i<=9;i++)
	{
	    for(j=l1+1;j<l2;j++)
	    {
		node[get(1,i)].sum[j-1]++;
	    }
	}
    }
}

inline void build()
{
    int i,j,p,q;
    for(i=0;i<=9;i++)
    {
	if(node[1].to[i])
	{
	    node[node[1].to[i]].fail=1;
	    que.push(node[1].to[i]);
	}
	else node[1].to[i]=1;
    }
    for(;!que.empty();)
    {
	q=que.front();
	que.pop();
	for(i=0;i<=l2;i++) node[q].sum[i]+=node[node[q].fail].sum[i];
	for(i=0;i<=9;i++)
	{
	    if(node[q].to[i])
	    {
		node[node[q].to[i]].fail=node[node[q].fail].to[i];
		que.push(node[q].to[i]);
	    }
	    else node[q].to[i]=node[node[q].fail].to[i];
	}
    }
    for(i=1;i<=tt;i++)
    {
	for(j=1;j<=n;j++)
	{
	    node[i].sum[j]+=node[i].sum[j-1];
	}
    }
}

int dfs(int now,int pos)
{
    if(now==n) return 0;
    if(dp[now][pos]!=-1) return dp[now][pos];
    int i,j,t,res=-1;
    for(i=0;i<=9;i++)
    {
	t=dfs(now+1,node[pos].to[i])+node[node[pos].to[i]].sum[n-now-1];
	if(t>res)
	{
	    to[now][pos]=i;
	    res=t;
	}
    }
    return dp[now][pos]=res;
}

int main()
{
    memset(dp,-1,sizeof(dp));
    int i,j,k,t;
    scanf("%s",a+1);l1=strlen(a+1);
    scanf("%s",b+1);l2=strlen(b+1);
    cin>>n;
    tt=1,in();
    build();
    cout<<dfs(0,1)<<endl;
    for(i=0,t=1;i<n;i++)
    {
	printf("%d",to[i][t]);
	t=node[t].to[to[i][t]];
    }
}