BZOJ4633 SCOI2016 萌萌哒 并查集维护ST表

题目大意

一个长度为$N$的数，$S_i$表示这个数的第i位是什么，现在题目给出了$M$个约束，每个约束为$lx,rx,ly,ry$，表示这个数的子串$S[lx...rx]$等于另一个子串$S[ly...ry]$。问满足题目约束的，长度为$N$的数有多少个。

$N \leq 10^5$
$M\leq 10^5$

解题思路

一个很显然但是会T的做法就是把一个约束拆成$rx - lx + 1$个约束，每个约束相当于在两个点之间连一条边。然后对于序列中的数维护一个并查集来记录有多少个联通块块，最后统计一下有多少个联通块。
我们发现有很多多余的操作，其实我们可以分层记录来避免多余的操作，对于某些块如果我们访问过了就不用再进行多余的操作了，其实这个可以用ST表的思想来维护，用$F[i][j]$表示区间$[i, i + 2^j - 1]$这个块在第j层的并查集情况，对于每个限制，我们可以拆成两个长度为$2^k$的区间来递归，并且把第j层的两个块合并。而当我们扫到之前已经合并过的块，那么就不用再递归进去了。
我们可以发现，对于每个$F[i][j]$，我们只会合并一次，所以时间复杂度是$O（NlogN）$，对于每次查询，我们就可以查询$F[i][0]$的合并情况。

程序

//YxuanwKeith
#include <cstring>
#include <cstdio>
#include <algorithm>

using namespace std;

const int MAXN = 1e5 + 5;
const int Mo = 1e9 + 7;

int N, M, Fa[MAXN], Log[MAXN], Pow[20], Bel[MAXN][20];
int lx, rx, ly, ry;

void Prepare() {
    Pow[0] = 1;
    for (int i = 1, j = 0; i <= N; i = 2 * i, j ++) Log[i] = j;
    for (int i = 1; i <= N; i ++) Log[i] = max(Log[i], Log[i - 1]);
    for (int i = 1; i <= 19; i ++) Pow[i] = Pow[i - 1] * 2;
    for (int i = 1; i <= N; i ++) {
        for (int j = 0; j <= Log[N]; j ++)
            Bel[i][j] = i;
    }
}

int Get(int Now, int Lg) {
    if (Bel[Now][Lg] == Now) return Now;
    Bel[Now][Lg] = Get(Bel[Now][Lg], Lg);
    return Bel[Now][Lg];
}

void Merge(int L1, int L2, int Lg) {
    int f1 = Get(L1, Lg), f2 = Get(L2, Lg);
    if (f1 == f2) return;
    Bel[f1][Lg] = f2;
    if (!Lg) return;
    Merge(L1, L2, Lg - 1), Merge(L1 + Pow[Lg - 1], L2 + Pow[Lg - 1], Lg - 1);
}

void Solve() {
    while (lx <= rx) {
        int Lg = Log[rx - lx + 1];
        Merge(lx, ly, Lg);
        lx = lx + Pow[Lg], ly = ly + Pow[Lg];
    }
}

int main() {
    scanf("%d%d", &N, &M);
    Prepare();
    for (int i = 1; i <= M; i ++) {
        scanf("%d%d%d%d", &lx, &rx, &ly, &ry);
        Solve();
    }

    int Cnt = 0, Ans = 1;
    for (int i = 1; i <= N; i ++) {
        if (Get(i, 0) == i) {
            Cnt ++;
            if (Cnt == 1) Ans = 1ll * Ans * 9 % Mo; else
                Ans = 1ll * Ans * 10 % Mo;
        }
    }
    printf("%d\n", Ans);
}