2021 ccpc桂林 J - Suffix Automaton(后缀自动机+线段树)

题目地址
赛中到最后时间不够,写的时候脑袋混乱,后来因为不明错误样例都没调过...没拿金我背大锅。
赛后重新整理下思路,直接A了,哎....
思路:首先对子串字典序问题有个经典做法是反串建后缀自动机,然后建出parent树,其中在建树时边权是子节点比父节点刚好多出一个的字符,这样这个树的dfs序就刚好是子串的字典序了,然后题目中是在子串长度相等的基础上比较字典序,否则比较长度,那么对于树上的某一个节点,由于其代表的子串长度是连续的,所以可以利用差分的思想,将这个子串的信息存到差分数组的该子串的最短长度和最长长度+1上,代表这段区间的长度含这个子串。然后就先按从小到大离线询问,然后对于每个询问,算出这个询问代表的子串的长度k和在这个长度里排第几x,然后将差分数组的指针一步步移到当前长度,一边移动一边更新线段树(将相应的dfs序更新),然后查询线段树中第x个出现的下标即可。由于最后答案要求是最左边出现的子串的下标,也就是要求反串后最右下标,可以从parent树从叶子往上更新位置的最大值。

越想越气...

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <vector>
#include <queue>
#include <set>
#include <stack>
#include <time.h>
#include <map>
#include <algorithm>
#include <fstream>
//#include <unordered_map>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const int maxn = 1000000 + 100;
const int INF = 0x7fffffff;
const int mod = 998244353;
const ll mod1 = 998244353;
const ll base = 137;
const double Pi = acos(-1.0);
const int G = 3;
struct state
{
    int len, link;
    //std::map<char, int> next;
    int nxt[26];
    int pos;
};
const int MAXLEN = 1000000 + 5;
state st[MAXLEN * 2];
int sz, last;
void sam_init()
{
    memset(st[0].nxt, 0, sizeof(st[0].nxt));
    sz = 0;
    st[0].len = 0;
    st[0].link = -1;
    sz++;
    last = 0;
}
int cur;
void sam_extend(char c, int id)
{
    cur = sz++;
    st[cur].pos = id;
    st[cur].len = st[last].len + 1;
    int p = last;
    while (p != -1 && !st[p].nxt[c - 'a'])
    {
        st[p].nxt[c - 'a'] = cur;
        p = st[p].link;
    }
    if (p == -1)
    {
        st[cur].link = 0;
    }
    else
    {
        int q = st[p].nxt[c - 'a'];
        if (st[p].len + 1 == st[q].len)
        {
            st[cur].link = q;
        }
        else
        {
            int clone = sz++;
            st[clone].len = st[p].len + 1;
            memcpy(st[clone].nxt, st[q].nxt, sizeof(st[clone].nxt));
          //  st[clone].pos = st[q].pos;
            //st[clone].next = st[q].next;
            st[clone].link = st[q].link;
            while (p != -1 && st[p].nxt[c - 'a'] == q)
            {
                st[p].nxt[c - 'a'] = clone;
                p = st[p].link;
            }
            st[q].link = st[cur].link = clone;
        }
    }
    last = cur;
    //siz[cur]=1;
}
int len[maxn];
int a[maxn*2];
int n;
void getSiz()
{
    for(int i=1;i<sz;i++)
    {
        len[st[i].len]++;
    }
    for(int i=n;i>=1;i--)
    {
        len[i]+=len[i+1];
    }
    for(int i=1;i<sz;i++)
    {
        a[len[st[i].len]--]=i;
    }
    for(int i=1;i<sz;i++)
    {
        int p=a[i];
        st[st[p].link].pos=max(st[st[p].link].pos,st[p].pos);
    }
}
char s[maxn];
struct node
{
    int id;
    ll x;
} qu[maxn];
bool cmp(node a, node b)
{
    return a.x < b.x;
}
struct NNode
{
    int to;
    int str;
    NNode(int tt = 0, int ss = 0)
    {
        to = tt;
        str = ss;
    }
};
vector<NNode> edge[maxn];
int in[maxn * 2];
int vis[maxn * 2];
int cnt;
void dfs(int x)
{
    in[x] = ++cnt;
    vis[cnt] = x;
    //cout<<222<<' '<<cnt<<' '<<x<<endl;
    for (int i = 0; i < (int)edge[x].size(); i++)
    {

        dfs(edge[x][i].to);
    }
}
struct pp
{
    int id;
    int f;
    pp(int ii, int ff)
    {
        id = ii;
        f = ff;
    }
};
vector<pp> v[maxn];
ll sum[maxn];
pair<int, int> ans[maxn];
struct Node
{
    int l, r, sum;
} tree[maxn * 8];
void build(int i, int l, int r)
{
    //cout<<i<<' '<<l<<' '<<r<<endl;
    tree[i].l = l;
    tree[i].r = r;
    if (l == r)
    {
        tree[i].sum = 0;
        return;
    }
    int mid = (l + r) >> 1;
    build(i * 2, l, mid);
    build(i * 2 + 1, mid + 1, r);
    tree[i].sum = 0;
}
void update(int i, int k, int x)
{
    if (tree[i].l == tree[i].r)
    {
        tree[i].sum += x;
        return;
    }
    if (tree[i * 2].r >= k)
        update(i * 2, k, x);
    if (tree[i * 2 + 1].l <= k)
        update(i * 2 + 1, k, x);
    tree[i].sum = tree[i * 2].sum + tree[i * 2 + 1].sum;
}
int query(int i, int k)
{
    //  cout<<i<<' '<<tree[i].l<<' '<<tree[i].r<<' '<<tree[i*2].sum<<' '<<k<<endl;
    if (tree[i].l == tree[i].r)
    {
        return tree[i].l;
    }
    int res;
    if (tree[i * 2].sum >= k)
        res = query(i * 2, k);
    else
        res = query(i * 2 + 1, k - tree[i * 2].sum);
    return res;
}
int dp[maxn];
bool cmp2(NNode a, NNode b)
{
    return a.str < b.str;
}
int main()
{
    scanf("%s", s + 1);
    int len = strlen(s + 1);
    n=len;
    for (int i = 1; i <= len / 2; i++)
    {
        swap(s[i], s[len - i + 1]);
    }
    sam_init();
    for (int i = 1; i <= len; i++)
        sam_extend(s[i], i);
    getSiz();
    //  printf("%s",s+1);
    for (int i = 1; i < sz; i++)
    {
        edge[st[i].link].push_back(NNode(i, s[st[i].pos - st[st[i].link].len] - 'a'));
        //  cout<<222<<' '<<st[i].pos<<' '<<st[st[i].link].len<<' '<<s[st[i].pos-st[st[i].link].len]-'a'<<endl;
    }
    for (int i = 0; i < sz; i++)
    {
        sort(edge[i].begin(), edge[i].end(), cmp2);
    }
    dfs(0);
    for (int i = 1; i < sz; i++)
    {
        v[st[st[i].link].len + 1].push_back(pp(i, 0));
        v[st[i].len + 1].push_back(pp(i, 1));
        dp[st[st[i].link].len + 1]++;
        dp[st[i].len + 1]--;
    }
    for (int i = 1; i <= len; i++)
    {
        sum[i] = sum[i - 1] + dp[i];
        //  cout<<111<<' '<<i<<' '<<sum[i]<<endl;
    }
    for (int i = 1; i <= len; i++)
    {
        sum[i] += sum[i - 1];
    }
    int q;
    scanf("%d", &q);
    for (int i = 1; i <= q; i++)
    {
        scanf("%lld", &qu[i].x);
        qu[i].id = i;
    }
    sort(qu + 1, qu + 1 + q, cmp);
    //cout<<len<<endl;
    build(1, 1, sz);
    int now = 1;
    for (int i = 1; i <= q; i++)
    {
        int le = lower_bound(sum + 1, sum + 1 + len, qu[i].x) - sum;
        int k = qu[i].x - sum[le - 1];
        //   cout<<le<<' '<<k<<endl;
        if (le > len)
        {
            ans[qu[i].id].first = -1;
            ans[qu[i].id].second = -1;
            continue;
        }
        while (now <= le)
        {
            for (auto j : v[now])
            {
                if (j.f == 0)
                {
                    update(1, in[j.id], 1);
                    // cout<<in[j.id]<<' '<<1<<endl;
                }
                else
                {
                    update(1, in[j.id], -1);
                }
            }
            now++;
        }
        int p = query(1, k);
        //  cout<<111<<' '<<k<<' '<<vis[p]<<endl;
        p=vis[p];
      //  while(st[st[p].link].len>=le) p=st[p].link;
        int l = st[p].pos - le + 1;
        int r = st[p].pos;
        ans[qu[i].id].first = len - r + 1;
        ans[qu[i].id].second = len - l + 1;
    }
    for (int i = 1; i <= q; i++)
    {
        printf("%d %d\n", ans[i].first, ans[i].second);
    }
    //  system("pause");
}
上一篇:element-UI相关问题


下一篇:Flume经典案例之自定义Source