【LG3247】[HNOI2016]最小公倍数

题面

洛谷

题解

50pts

因为拼凑起来的部分分比较多，所以就放一起了。

以下设询问的\(a,b\)为\(A,B\)，

复杂度\(O(nm)\)的：将所有\(a\leq A,b\leq B\)的边两端，用并查集并起来，再看一看等于\(A,B\)的是否有端点在集合中即可。

一条链的：拿线段树之类的数据结构维护一下即可。

\(a\)等于\(0\)的：将边的和询问按照\(b\)排序，用\(two\;pointers\)扫一遍丢到并查集中即可。

100pts

首先考虑暴力，每次询问暴力求出所有\(\leq a, \leq b\)的边，然后判断判断两点是否联通，并且联通块内最大值是否合法就可以了。

接下来的\(A\)和\(B\)还是是询问的\(a, b\)。

将所有的边按照\(a\)排序并分块，将所有的询问按照\(b\)排序。

设第\(i\)块的区间是\([l_i, r_i]\)，找出所有的\(A \in [a_{l_i}, a_{r_i})\)的询问，然后一个一个处理。

对于第\(j\)个询问，有两种边可以产生贡献，一种是在\([1, i)\)的\(b \leq B_j\)的边，这种边可以用一个指针维护。

还有一种是在第\(i\)块的\(a \leq A_j\)，\(b \leq B_j\)的边，这种边最多只有块的大小条，可以暴力加边。

因为每一次加了第二种边之后要撤销这些操作，所以要写一个支持撤销的并查集。

然后，当块的大小为\(\sqrt{m\log_2n}\)时据说最快。

(感谢\(xgzc\)友情提供)

代码

#include <iostream>

#include <cstdio>

#include <cstdlib>

#include <cstring>

#include <cmath>

#include <algorithm>

using namespace std;

inline int gi() {

    register int data = 0, w = 1;

    register char ch = 0;

    while (!isdigit(ch) && ch != '-') ch = getchar();

    if (ch == '-')

        w = -1, ch = getchar();

    while (isdigit(ch)) data = 10 * data + ch - '0', ch = getchar();

    return w * data;

}

const int MAX_N = 2e5 + 5;

struct Edge { int u, v, a, b; } e[MAX_N];

struct Query { int u, v, a, b, id; } q[MAX_N], p[MAX_N];

struct Node { int u, v, a, b, s; } stk[MAX_N];

bool cmp1(const Edge &l, const Edge &r) { return l.a == r.a ? l.b < r.b : l.a < r.a; }

bool cmp2(const Edge &l, const Edge &r) { return l.b == r.b ? l.a < r.a : l.b < r.b; }

bool cmp3(const Query &l, const Query &r) { return l.b == r.b ? l.a < r.a : l.b < r.b; }

int N, M, Q, top, ans[MAX_N], fa[MAX_N], A[MAX_N], B[MAX_N], size[MAX_N];

int getf(int x) { return fa[x] == x ? x : getf(fa[x]); }

void merge(int x, int y, int a, int b) {

    x = getf(x), y = getf(y);

    if (size[x] > size[y]) swap(x, y);

    stk[++top] = (Node){x, y, A[y], B[y], size[y]};

    if (x != y)

        fa[x] = y, size[y] += size[x], A[y] = max(A[x], A[y]), B[y] = max(B[x], B[y]);

    A[y] = max(A[y], a);

    B[y] = max(B[y], b);

}

int main() {

#ifndef ONLINE_JUDGE

    freopen("cpp.in", "r", stdin);

#endif

    N = gi(), M = gi();

    for (int i = 1; i <= M; i++) e[i] = (Edge){gi(), gi(), gi(), gi()};

    Q = gi();

    for (int i = 1; i <= Q; i++) q[i] = (Query){gi(), gi(), gi(), gi(), i};

    sort(&e[1], &e[M + 1], cmp1);

    sort(&q[1], &q[Q + 1], cmp3);

    for (int i = 1, LEN = sqrt(M * log2(N)); i <= M; i += LEN) {

        for (int j = 1; j <= N; j++) size[fa[j] = j] = 1, A[j] = B[j] = -1;

        int tot = 0;

        for (int j = 1; j <= Q; j++)

            if (e[i].a <= q[j].a && (i + LEN > M || q[j].a < e[i + LEN].a))

                p[++tot] = q[j];

        if (!tot) continue;

        sort(&e[1], &e[i], cmp2);

        for (int j = 1, k = 1; j <= tot; j++) {

            while (k < i && e[k].b <= p[j].b) merge(e[k].u, e[k].v, e[k].a, e[k].b), ++k;

            top = 0;

            for (int l = i; l < i + LEN && l <= M; l++)

                if (e[l].a <= p[j].a && e[l].b <= p[j].b)

                    merge(e[l].u, e[l].v, e[l].a, e[l].b);

            int x = getf(p[j].u), y = getf(p[j].v);

            ans[p[j].id] = (x == y && A[x] == p[j].a && B[x] == p[j].b);

            while (top) {

                int x = stk[top].u, y = stk[top].v;

                fa[x] = x;

                A[y] = stk[top].a, B[y] = stk[top].b, size[y] = stk[top].s;

                --top;

            }

        }

    }

    for (int i = 1; i <= Q; i++) puts(ans[i] ? "Yes" : "No");

    return 0;

}