2019南京ICPC（重现赛） F - Paper Grading

题目链接：https://nanti.jisuanke.com/t/42400

这还是去年去现场赛打的，当时菜的不行，就白给了。最近学了主席树套树状数组，感觉好强的数据结构啊。我们学长说这题挺简单，建字典树dfs序，跑cdq分治就好了（%%%）。本菜鸡发现这题主席树套树状数组也能做。

 

题意：给你n个字符串，m个操作。操作1交换俩个字符串，操作2，给定一个字符串s，数字k, l, r， 求l ~ r与s公共前缀大于等于k的字符串的个数。

 

首先考虑没有修改，那么就只有操作2，设f[i]为第i个字符串结尾的dfn序。把要询问的字符串在字典树里面第k的位置x求出来。那么就变成求in[x]~out[x]中存在的f[l] ~ f[r]数量，考虑是区间询问，所以用主席树。我们在每个in[x]上建值域线段树（1， n）并在i上加1。就可以变成求in[x]~out[x]中存在l~r的数量，很明显的主席树。然后考虑修改，每次修改会把f[i]和f[j]的位置交换，所以就交换了值域i, j。

 

#include <bits/stdc++.h>

using namespace std;

const int N = 2e5 + 10;

int n, m;

char s[N];

int f[N];

struct Trie{
    int son[N][26], idx = 0;
    int in[N], out[N], num = 0;
    int insert(char *s)
    {
        int p = 0;
        for (int i = 0; s[i]; i ++)
        {
            int u = s[i] - 'a';
            if(!son[p][u])
            son[p][u] = ++ idx;
            p = son[p][u];
        }
        return p;
    } 
    void dfs(int x)
    {
        in[x] = ++ num;
        for (int i = 0; i < 26; i ++)
            if(son[x][i])
                dfs(son[x][i]);
        out[x] = num;
    }
    int find(char* s, int k)
    {
        int p = 0;
        for (int i = 0; i < k; i ++)
        {
            int u = s[i] - 'a';
            if(!son[p][u])
            return -1;
            p = son[p][u];
        }
        return p;
    }
}t;

struct Seg{
    int l, r;
    int val;
}tr[N * 100];

int root[N];
int num;

void build(int &rt, int l, int r)
{
    if(!rt)
    rt = ++ num;
    if(l == r) return ;
    int mid = l + r >> 1;
    build(tr[rt].l, l, mid);
    build(tr[rt].r, mid + 1, r); 
}

void update(int &p, int x, int c, int l, int r)
{
    if(!p) p = ++ num;
    tr[p].val += c;
    if(l == r) return ;
    int mid = l + r >> 1;
    if(x <= mid)
    update(tr[p].l, x, c, l, mid);
    else
    update(tr[p].r, x, c, mid + 1,r);
}

int lowbit(int x)
{
    return x & (-x);
}
void update(int x, int y, int c)
{
    for (; x < N; x += lowbit(x))
    {
        update(root[x], y, c, 1, n);
    }
}

int query(int p, int L, int R, int l, int r)
{
    if(L <= l && R >= r)
    {
        return tr[p].val;
     } 
     int mid = l + r >> 1;
        int res = 0;
    if(L <= mid)
    res += query(tr[p].l, L, R, l, mid);
    if(R >  mid)
    res += query(tr[p].r, L, R, mid + 1, r);
    return res;     
}

int query(int x, int y, int L, int R)
{
    int res = 0;
    for (; y; y -= lowbit(y))
        res += query(root[y], L, R, 1, n);
    for (; x; x -= lowbit(x))
        res -= query(root[x], L, R, 1, n);
    return res;
}

int main()
{
    scanf("%d%d", &n, &m);
    for (int i = 1; i <= n; i ++)
    {
        scanf("%s",s);
        f[i] = t.insert(s);
    }
    t.dfs(0);
    build(root[0], 1, n);
    for (int i = 1; i <= n; i ++)
    {
        update(t.in[f[i]], i, 1);
        f[i] = t.in[f[i]];
    }
    while(m --)
    {
        int op;
        scanf("%d", &op);
        if(op == 1)
        {
            int l, r;
            scanf("%d%d", &l, &r);
            update(f[l], l, -1);
            update(f[r], r, -1);
            swap(f[l], f[r]);
            update(f[l], l, 1);
            update(f[r], r, 1);
        }
        else
        {
            int k, l, r;
            scanf("%s%d%d%d", s, &k, &l, &r);
            int pos = t.find(s, k);
            if(pos == -1)
            {
                puts("0");
                continue;
            }
            int L = t.in[pos] - 1, R = t.out[pos];
        //    cout << L << " " << R << endl;
            printf("%d\n", query(L, R, l, r));
        }
    }
}