1. 程式人生 > >[NOI2017]蚯蚓排隊 題解

[NOI2017]蚯蚓排隊 題解

題目就是要求我們能夠支援三個操作:

1.把兩個字串合併

2.把兩個字串分裂

3.詢問有多少一些個串是這麼多串中出現了多少次

我們發現數據範圍中的 K K 很小,於是我們就考慮出題人為什麼 K K 這麼小

這樣的話需要匹配的字串的長度最長就只有 K K

於是我們發現第3中操作可以通過雜湊表解決,雜湊表只需要儲存長度 < = K <=K

的子串的雜湊值即可(雜湊表就是掛鏈就行了)

我們思考,兩個字串 a , b a,b 合併會增加哪些雜湊值? 其實就是那些即有 a a

的部分又有 b b 部分的子串。 這樣的串有多少個?可能有很多,但是我們記錄的最多有 K 2 K^2 個。

分裂?不也是一樣的麼,就是把那些即有 a a 的部分又有 b b 部分的子串的雜湊值從雜湊表中移除

對於第三種,只需要 O ( S ) O(|S|) 的列舉串一遍,然後滾動雜湊一下不就行了麼

有應為 S < = 1 0 7 \sum{|S}|<=10^7 ,所以總時間複雜度大概就是 O ( 1 0 7 + M K 2 ) = > O ( M K 2 ) O(10^7+MK^2)=>O(MK^2)

最好用讀入優化,防止TLE

打程式碼的難度還是有的,大家加油吧

程式碼如下

#include <bits/stdc++.h>
using namespace std ;
#define rep(i, a, b) for (int (i) = (a); (i) <= (b); (i)++)
#define per(i, a, b) for (int (i) = (a); (i) >= (b); (i)--)
#define clr(a) memset(a, 0, sizeof(a))
#define ass(a, sum) memset(a, sum, sizeof(a))
#define lowbit(x) (x & -x)
#define mp make_pair
#define pb push_back
#define fi first
#define se second
#define enter cout << endl
#define siz(x) ((int)x.size())
typedef long long ll ;
typedef unsigned long long ull ;
typedef vector <int> vi ;
typedef pair <int, int> pii ;
typedef map <int, int> mii ;
typedef map <string, int> msi ;
const int N = 200010 ;
const int M = 20000000 ;
const int P = 10000009 ;
const int K = 50 ;
const int INF = 0x3f3f3f3f ;
const int iinf = 1 << 30 ;
const ll linf = 2e18 ;
const int MOD = 998244353 ;
const ull bs = 7 ;
void print(int x) { cout << x << endl ; exit(0) ; }
void PRINT(string x) { cout << x << endl ; exit(0) ; }
void douout(double x){ printf("%lf\n", x + 0.0000000001) ; }

namespace io {
    const int BUFSIZE = 1 << 20 ;
    char ibuf[BUFSIZE], *is = ibuf, *it = ibuf ;
    char readc() {
        if (is == it) it = (is = ibuf) + fread(ibuf, 1, BUFSIZE, stdin) ;
        return *is++ ;
    }
    int readi() {
        int x = 0, f = 1 ;
        char c = readc() ;
        while (!isdigit(c)) {
            if (c == '-') f = -1 ;
            c = readc() ;
        }
        while (isdigit(c)) {
            x = x * 10 + c - '0' ;
            c = readc() ;
        }
        return x * f ;
    }
}

using io::readc ;
using io::readi ;

struct Hash_table {
    int tot, head[P], nxt[M], val[M] ; ull to[M] ;
    void modify(ull x, int sum) {
        int ind = x % P ;
        for (int i = head[ind]; i; i = nxt[i]) // link
        if (to[i] == x){ // appeared
            val[i] += sum ;
            return ;
        }
        to[++tot] = x ; // first appear, push into the hash table
        nxt[tot] = head[ind] ;
        val[tot] = sum ;
        head[ind] = tot ;
    }
    int query(ull x) {
        for (int i = head[x % P]; i; i = nxt[i]) if (to[i] == x) return val[i] ; // find
        return 0 ; // not find
    }
} Hash ;

int n, m ;
int pre[N], suf[N], str[N], s[M] ;
ull pw[N] ;

signed main(){
    n = readi() ; m = readi() ;
    for (int i = 1; i <= n; i++) {
        str[i] = readi() ;
        Hash.modify(str[i], 1) ;
    }
    pw[0] = 1 ; for (int i = 1; i <= K; i++) pw[i] = pw[i - 1] * bs ;
    while (m--) {
        int ty = readi(), k ;
        if (ty == 1) {
            int x = readi(), y = readi() ; 
            ull now = 0 ;
            pre[y] = x, suf[x] = y ; // change link
            for (int i = x, p = 1; i && p <= K; i = pre[i], p++) {
                ull hqg = now += pw[p - 1] * str[i] ;
                for (int j = y, q = p + 1; j && q <= K; j = suf[j], q++) {
                    hqg = hqg * bs + str[j] ;
                    Hash.modify(hqg, 1) ;
                }
            }
        } else
        if (ty == 2) {
            int x = readi(), y ; y = suf[x] ;
            ull now = 0 ;
            pre[y] = suf[x] = 0 ;
            for (int i = x, p = 1; i && p <= K; i = pre[i], p++) {
            	ull hqg = now += pw[p - 1] * str[i] ;
            	for (int j = y, q = p + 1; j && q <= K; j = suf[j], q++) {
                    hqg = hqg * bs + str[j] ;
                    Hash.modify(hqg, -1) ;
                }
            }
        } else {
            char c = readc() ;
            while (!isdigit(c)) c = readc() ;
            int len = 0 ;
            while (isdigit(c)) s[len++] = c - '0', c = readc() ;
            k = readi() ;
            ull now = 0, ans = 1 ;
            for (int i = 0; i < len; i++) {
                now = now * bs + s[i] ;
                if (i >= k - 1) {
                    ans = ans * Hash.query(now) % MOD ;
                    now -= pw[k - 1] * s[i - k + 1] ;
                }
            }
            printf("%llu\n", ans) ;
        }
    }
    return 0 ;
}