2019 ICPC 南京 F题 Paper Grading

题目链接

题意

给一些字符串，一个操作是交换两个串的位置，另一个是问编号介于 $[l,r]$ 的字符串，与提问串的最长公共前缀（LCP）至少为 $k$ 的个数

题解

将串建立 $Trie$ 树，结点记录所属的串的位置，考虑交换操作，就交换 $Trie$ 中维护的位置信息就好了，提问时，从根开始走 $k$ 步，最后到达结点的子树都是符合条件的，然后求一下子树中，位置介于 $[l,r]$ 的个数即可

用主席树（带修改）

注意的是交换操作
提问的是子串的标号，所以需要维护，当前编号的字符串，最初的编号，然后再映射到 $Trie$ 的 $dfs$ 序中，然后再进行交换操作
还有一个坑点，有重复的子串，所以一个结点可能有多个位置信息，用 $vector$ 存

代码

#include<bits/stdc++.h>
#define N 200010
#define INF 0x3f3f3f3f
#define eps 1e-8
#define pi 3.141592653589793
#define mod 998244353
#define P 1000000007
#define LL long long
#define pb push_back
#define fi first
#define se second
#define cl clear
#define si size
#define lb lower_bound
#define ub upper_bound
#define bug(x) cerr<<#x<<"      :   "<<x<<endl
#define mem(x,y) memset(x,0,sizeof(int)*(y+3))
#define sc(x) scanf("%d",&x)
#define scc(x,y) scanf("%d%d",&x,&y)
#define sccc(x,y,z) scanf("%d%d%d",&x,&y,&z)
using namespace std;
typedef pair<int,int> pp;
int rt[N],f[N],tot;
int n,m;
int w[N],in[N],out[N],cnt;
char x[N];
struct PST{
    int ls[N*200],rs[N*200],sum[N*200],l1[20],l0[20],p1,p0,cnt,S,t,w[2][20][20],p[2];
    void init(int x,int Max){S=x;cnt=0;tot=Max;}
    inline void ins(int &i,int x,int val,int l=1,int r=tot){
        if(!i) i=++cnt; int j=i;
        while(l<r){
            sum[j]+=val;
            int mid=l+r>>1;
            if (x<=mid) r=mid,j=ls[j]?ls[j]:ls[j]=++cnt;
            else l=mid+1,j=rs[j]?rs[j]:rs[j]=++cnt;
        }
        sum[j]+=val;
    }
    //将位置pos的值删去/加上，val=-1/1
    inline void Modify(int pos,int x,int val){
        for(int i=pos;i<=S;i+=i&-i) ins(rt[i],x,val);
    }
    //int w[0][20][20],p[2];
    int ask(int x,int y,int d,int l=1,int r=tot){
        int v=d+1;
        if (l==x&&y==r){
            int s=0; for(int i=1;i<=t;i++) s+=sum[w[1][d][i]],s-=sum[w[0][d][i]];
            return s;
        }
        int mid=l+r>>1;
        if (y<=mid) {
            for(int i=1;i<=t;i++) w[0][v][i]=ls[w[0][d][i]],w[1][v][i]=ls[w[1][d][i]];
            return ask(x,y,v,l,mid);
        }else
        if (x>mid) {
            for(int i=1;i<=t;i++) w[0][v][i]=rs[w[0][d][i]],w[1][v][i]=rs[w[1][d][i]];
            return ask(x,y,v,mid+1,r);
        }else{
            for(int i=1;i<=t;i++) w[0][v][i]=ls[w[0][d][i]],w[1][v][i]=ls[w[1][d][i]];
            int t1=ask(x,mid,v,l,mid);
            for(int i=1;i<=t;i++) w[0][v][i]=rs[w[0][d][i]],w[1][v][i]=rs[w[1][d][i]];
            int t2=ask(mid+1,y,v,mid+1,r);
            return t1+t2;
        }
    }
    //区间[l,r]范围为数值在[x,y]的个数
    inline int query(int l,int r,int x,int y){
        p[0]=p[1]=0;memset(w,0,sizeof w);
        for(int i=l-1;i;i-=i&-i) w[0][0][++p[0]]=rt[i];
        for(int i=r;i;i-=i&-i) w[1][0][++p[1]]=rt[i];
        t=max(p[0],p[1]);
        return ask(x,y,0);
    }
}PST;


struct Trie{
    int t[N][26],pos[N],a[N],tot,num; vector<int>  mp[N];
    void ins(char* x,int cc){
        int i=0; 
        for(int j=0;x[j];j++){
            int c=x[j]-'a';
            if(!t[i][c]) t[i][c]=++tot;
            i=t[i][c];
        }
        mp[i].pb(cc);
        w[i]=cc;
    }
    void dfs(int x){
        in[x]=++cnt; 
        for(auto i:mp[x]) pos[i]=++cnt,a[cnt]=i+1;
        for(int i=0;i<26;i++) if (t[x][i]) dfs(t[x][i]);
        out[x]=cnt;
    }
    void build(){
        dfs(0);
        PST.init(cnt,n+1);
        for(int i=1;i<=cnt;i++) PST.Modify(i,a[i],1);
    }
    void sol(char *x,int k,int l,int r){
        int rt=0;
        for(int i=0;i<k;i++){
            int c=x[i]-'a';
            if (t[rt][c]) rt=t[rt][c];else{
                puts("0");return;
            } 
        }
        printf("%d\n",PST.query(in[rt],out[rt],l+1,r+1));
    }
    void change(int x,int y){
        int l=f[x],r=f[y]; swap(f[x],f[y]); 
        l=pos[l],r=pos[r];
        PST.Modify(l,a[l],-1);
        PST.Modify(r,a[r],-1);
        swap(a[l],a[r]);
        PST.Modify(l,a[l],1);
        PST.Modify(r,a[r],1);
    }
}T;

int main(int argc, char const *argv[])
{
    scc(n,m);
    for(int i=1;i<=n;i++) scanf("%s",x), T.ins(x,i), f[i]=i;
    T.build();
    while(m--){
        int op; sc(op);
        if (op==2){
            int k,l,r;
            scanf("%s",x); sccc(k,l,r);
            T.sol(x,k,l,r);
        }else{
            int l,r; scc(l,r);
            T.change(l,r);
        }
    }
    return 0;
}