2019年华南理工大学程序设计竞赛 华工校赛 K Parco_Love_String（dp/后缀自动机SAM板子）

题目链接

题目大意，给你一个长度为1e3的字符串S，1e5次询问。每次询问给你一个k，问与的相同子串对数。

• 显然可以用后缀数组和后缀自动机先预处理答案去做，赛中不会打。大概排名附近的除了我们队全都A了。。。
• 赛后在牛客看代码的时候发现了一个新的思路，理解后想要分享一下。
  • 首先我们可以用的时间预处理出所有i,j的与的所有后缀的相同的对数
  • 同理，也可以预处理出所有i,j的和所有前缀的相同的对数
  • 当我们知道ans[i-1]的条件下，假设为X
  • 我们把第i个后缀与第i+1~|S|个后缀相同的对数的贡献到X
  • 并把第i个前缀与第1~i-1个前缀相同的对数的贡献删去
  • 明显的X就是我们所需要的ans[i]
  • 然后这个转移方程明显是的，总复杂度为

dp代码

#include<bits/stdc++.h>

#define qcin; ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#define pb push_back
#define mp make_pair
#define clr(x) memset(x,0,sizeof x)
#define fmax(x) memset(x,0x3f,sizeof x)
#define finit(x) memset(x,-1,sizeof x)
#define VI vector<int>
#define dis(l,r) r-l+1
#define DIS(x,y) sqrt((x.first-y.first)*(x.first-y.first)+(x.second-y.second)*(x.second-y.second))

#define gstr(str) scanf("%s",str)
#define glen(str) strlen(str)
#define ls(x) (x<<1)
#define rs(x) ((x<<1)|1)

using namespace std;

typedef long long ll;
typedef pair<ll,ll > pii;
const int maxn = 1e3+10;
const double eps = 1e-10;
const ll INF = 1e15;

void file(int x){if(x&&fopen("123.in","r")){freopen("123.in","r",stdin);}}

const double delta=0.993;

char s[maxn];
int f[maxn][maxn],g[maxn][maxn],res[maxn];
int t,n,m,k;

int main(){
    file(1);
    gstr(s+1);
    n=glen(s+1);
    for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++){
            if(s[i]==s[j])f[i][j]=f[i-1][j-1]+1;
        }
    }
    for(int i=n;i;i--){
        for(int j=n;j;j--){
            if(s[i]==s[j])g[i][j]=g[i+1][j+1]+1;
        }
    }
    for(int i=2;i<=n;i++)if(s[i]==s[1])res[1]++;

    for(int i=2;i<=n;i++){
        res[i]=res[i-1];
        for(int j=i+1;j<=n;j++){
            res[i]+=min(j-i,f[i][j]);
        }
        for(int j=1;j<i;j++){
            res[i]-=min(i-j,g[i][j]);
        }
    }
    for(scanf("%d",&t);t--;){
        scanf("%d",&k);
        printf("%d\n",res[k]);
    }
}

SAM代码

• 整体思路是维护出每个节点endpos的大小（sz），然后做一次前缀和，对于每个点维护出的以当前点所代表的的字符串的后缀出现的次数（sum）。
• 然后对着另一个串跑一下SAM就可以得到结果了

#include<bits/stdc++.h>

#define qcin; ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#define pb push_back
#define mp make_pair
#define clr(x) memset(x,0,sizeof x)
#define fmax(x) memset(x,0x3f,sizeof x)
#define finit(x) memset(x,-1,sizeof x)
#define VI vector<int>
#define dis(l,r) r-l+1
#define DIS(x,y) sqrt((x.first-y.first)*(x.first-y.first)+(x.second-y.second)*(x.second-y.second))

#define gstr(str) scanf("%s",str)
#define glen(str) strlen(str)
#define ls(x) (x<<1)
#define rs(x) ((x<<1)|1)

using namespace std;

typedef long long ll;
typedef pair<ll,ll > pii;
const int maxn = 2e3+10;
const double eps = 1e-10;
const ll INF = 1e15;
const int C_sz= 26;
void file(int x){if(x&&fopen("123.in","r")){freopen("123.in","r",stdin);}}

const double delta=0.993;

int las,tot,root=1;

char s[maxn];
ll res[maxn],sum[maxn],sz[maxn];
int num[maxn],id[maxn],n,t,k;

struct sam_node{
    int ch[C_sz];
    int len,par;
    sam_node(int _len=0):
        par(0),len(_len){
        clr(ch);
    }
}SAM[maxn];

inline void extend(int c){
    int p=las,np=++tot;
    SAM[np]=sam_node(SAM[p].len+1);
    sz[np]=1;
    while(p&&!SAM[p].ch[c])SAM[p].ch[c]=np,p=SAM[p].par;
    if(!p)SAM[np].par=root;
    else{
        int q=SAM[p].ch[c];
        if(SAM[p].len+1==SAM[q].len){
            SAM[np].par=q;
        }else{
            int nq=++tot;
            SAM[nq]=SAM[q];
            SAM[nq].len=SAM[p].len+1;
            SAM[q].par=SAM[np].par=nq;
            while(p&&SAM[p].ch[c]==q)SAM[p].ch[c]=nq,p=SAM[p].par;
        }
    }
    las=np;
}

void topo(){
    for(int i=1;i<=tot;i++)num[SAM[i].len]++;
    for(int i=1;i<=tot;i++)num[i]+=num[i-1];
    for(int i=tot;i;i--)id[num[SAM[i].len]--]=i;
    for(int i=tot;i;i--){
        int x=id[i];int fa=SAM[x].par;
        sz[SAM[x].par]+=sz[x];
    }
    for(int i=1;i<=tot;i++){
        int x=id[i];int fa=SAM[x].par;
        sum[x]=sum[fa]+1LL*sz[x]*(SAM[x].len-SAM[fa].len);
    }
}

void cal(int x){
    las=1;tot=1;
    SAM[1]=sam_node();
    clr(sz);clr(num);
    ll ans=0;
    for(int i=1;i<=x;i++)extend(s[i]-'a');
    topo();
    for(int i=x+1,p=1,len=0;i<=n;i++){
        int c=s[i]-'a';
        if(SAM[p].ch[c]){
            ++len;
            p=SAM[p].ch[c];
        }else{
            while(p&&!SAM[p].ch[c])p=SAM[p].par;
            if(p){
                len=SAM[p].len+1;
                p=SAM[p].ch[c];
            }else p=1,len=0;
        }
        ans+=sum[SAM[p].par]+1LL*sz[p]*(len-SAM[SAM[p].par].len);
    }
    res[x]=ans;
}

int main(){
    file(1);
    gstr(s+1);
    n=glen(s+1);
    for(int i=1;i<=n;i++)cal(i);
    for(scanf("%d",&t);t--;){
        scanf("%d",&k);
        printf("%lld\n",res[k]);
    }
}