Address

Solution

对于 $1\leq i\leq m$ ，考虑分别预处理经度在 $[1,i]$ ， $[i,m]$ 的点的 $\text{MST}$ 。询问的时候合并 $[1,l-1]$ 和 $[r+1,m]$ 即可。

先考虑怎么预处理 $[1,i]$ 的 $\text{MST}$ （ $[i,m]$ 同理）。

假设我们已经有了 $[1,i-1]$ 的 $\text{MST}$ ，现在要在这里加上经度为 $i$ 的点和一些边。

考虑加入一条边 $(u,v,w)$ 会发生什么：

$1.$ $u,v$ 不连通，连接 $u,v$ 。
$2.$ $u,v$ 已经连通，且路径 $u→v$ 的边的最大值 $\leq w$ ，什么也不会发生。
$3.$ $u,v$ 已经连通，且路径 $u→v$ 的边的最大值 $> w$ ，断开这条最大边，连接 $u,v$ 。

也就是说， $[1,i-1]$ 的 $\text{MST}$ 中可能会有一些边被删掉。我们把 $\text{MST}$ 中第一列和最后一列的点称为关键点，那么被删掉的边 $l$ 必定满足：存在关键点 $x,y$ 使得 $l$ 是路径 $x→y$ 上的权值最大边。

换句话说，把端点的经度在 $[1,i-1]$ 的边全部拿出来跑 $\text{kruscal}$ 。边 $l$ 会连接两个连通块，如果这两个连通块里面都有关键点，那么 $l$ 可能被删掉，否则 $l$ 不可能被删掉。

因此记录边集 $pre_i$ 表示经度在 $[1,i]$ 的点的 $\text{MST}$ 中，之后可能被删的边。

$[1,i-1]$ 的 $\text{MST}$ 中，不在 $pre_{i-1}$ 的边（之后肯定不会被删的边）肯定都在 $[1,i]$ 的 $\text{MST}$ 中。假设 $pre_{i-1}$ 中的边的端点都是关键点，那么我们只要把 $pre_{i-1}$ 和新加入的 $2n-1$ 条边一起拿出来跑 $\text{kruscal}$ ，所得结果加上肯定不会被删的边，就是 $[1,i]$ 的 $\text{MST}$ 。

可是 $pre_{i-1}$ 的边的端点不一定都是关键点，怎么办呢？

考虑 $\text{kruscal}$ 的过程：

int fu = find(u), fv = find(v);
if (fu != fv) 在 MST 中加入边 (u, v, w);

其实和这样是等价的：

int fu = find(u), fv = find(v);
if (fu != fv) 在 MST 中加入边 (fu, fv, w);

假设 $pre_{i-1}$ 中的边都是关键点，那么可以考虑这样求出 $pre_i$ ：

inline void solve(vector<edge> &a, vector<edge> &b, ll &del)
{
    int len = a.size(), i;
    sort(a.begin(), a.end(), cmp);
    b.clear(); del = 0;
    for (i = 0; i < len; i++)
    {
        int x = a[i].x, y = a[i].y, v = a[i].v, fx = find(x), fy = find(y);
        if (fx == fy) del += v;
        else
        {
            if (bo[fx] && bo[fy]) link(fx, fy), b.push_back((edge){fx, fy, v}); 
            // 保证 b 中的边都是关键点
            else if (bo[fx]) link(fx, fy);
            else link(fy, fx);
        }
    }
}

其中 $a$ 是 $pre_{i-1}$ 加上新的 $2n-1$ 条边， $bo_x=1$ 表示 $x$ 是关键点， $b$ 是 $pre_i$ 。

这样我们就得到了 $pre_i$ ， $suf_i$ 同理。

对于询问 $(l,r)$ ，只要把 $pre_{l-1}$ 和 $suf_{r+1}$ 合并即可，方法跟已知 $pre_{i-1}$ 求 $pre_i$ 差不多。

时间复杂度 $O(nm\log n)$ 。

Code

#include <bits/stdc++.h>

using namespace std;

#define ll long long

template <class t>
inline void read(t & res)
{
    char ch;
    while (ch = getchar(), !isdigit(ch));
    res = ch ^ 48;
    while (ch = getchar(), isdigit(ch))
    res = res * 10 + (ch ^ 48);
}

template <class t>
inline void print(t x)
{
    if (x > 9) print(x / 10);
    putchar(x % 10 + 48);
}

const int N = 105, M = 10005, T = N * M;
struct edge
{
    int x, y, v;
};
vector<edge> pre[M], suf[M];
int n, rht[N][M], dwn[N][M], m, f[T], lim, q;
bool bo[T];
unsigned int SA, SB, SC; 
ll pres[M], sufs[M], ans; 
// pres[i] 表示 [1,i] 的 MST 的边权之和，sufs[i] 同理

inline int getweight() 
{
    SA ^= SA << 16;
    SA ^= SA >> 5;
    SA ^= SA << 1;
    unsigned int t = SA;
    SA = SB;
    SB = SC;
    SC ^= t ^ SA;
    return SC % lim + 1;
}

inline void gen() 
{
    read(n); read(m); read(SA); read(SB); read(SC); read(lim);
    int i, j;
    for (i = 1; i <= n; i++)
        for (j = 1; j <= m; j++) 
            rht[i][j] = getweight();
    for (i = 1; i < n; i++)
        for (j = 1; j <= m; j++) 
            dwn[i][j] = getweight();
}

inline int id(int x, int y)
{
    return (x - 1) * m + y;
}

inline int find(int x)
{
    return f[x] == x ? x : f[x] = find(f[x]);
}

inline bool cmp(const edge &a, const edge &b)
{
    return a.v < b.v;
}

inline void link(int x, int y)
{
    f[y] = x;
}

inline void upt(int x, bool y)
{
    f[x] = x;
    bo[x] = y;
}

inline void solve(vector<edge> &a, vector<edge> &b, ll &del)
{
    int len = a.size(), i;
    sort(a.begin(), a.end(), cmp);
    b.clear(); del = 0;
    for (i = 0; i < len; i++)
    {
        int x = a[i].x, y = a[i].y, v = a[i].v, fx = find(x), fy = find(y);
        if (fx == fy) del += v;
        else
        {
            if (bo[fx] && bo[fy]) link(fx, fy), b.push_back((edge){fx, fy, v});
            else if (bo[fx]) link(fx, fy);
            else link(fy, fx);
        }
    }
}

inline void init_pre()
{
    int i, j;
    for (i = 1; i <= n * m; i++) f[i] = i;
    ll del;
    vector<edge> a, b;
    for (i = 1; i <= m; i++)
    {
        for (j = 1; j <= n; j++)
        {
            upt(id(j, i), 1);
            upt(id(j, 1), 1);
            if (i > 2) upt(id(j, i - 1), 0);
        }
        a = pre[i - 1];
        pres[i] = pres[i - 1];
        for (j = 1; j <= n; j++)
        {
            if (i != 1)
            {
                a.push_back((edge){id(j, i - 1), id(j, i), rht[j][i - 1]});
                pres[i] += rht[j][i - 1];
            } 
            if (j != n)
            {
                a.push_back((edge){id(j, i), id(j + 1, i), dwn[j][i]});
                pres[i] += dwn[j][i];
            }
        }
        solve(a, b, del);
        pre[i] = b;
        pres[i] -= del;
    }
}

inline void init_suf()
{
    int i, j;
    ll del;
    for (i = 1; i <= n * m; i++) f[i] = i, bo[i] = 0;
    vector<edge> a, b;
    for (i = m; i >= 1; i--)
    {
        for (j = 1; j <= n; j++)
        {
            upt(id(j, i), 1);
            upt(id(j, m), 1);
            if (i < m - 1) upt(id(j, i + 1), 0);
        }
        a = suf[i + 1];
        sufs[i] = sufs[i + 1];
        for (j = 1; j <= n; j++)
        {
            if (i != m)
            {
                a.push_back((edge){id(j, i + 1), id(j, i), rht[j][i]});
                sufs[i] += rht[j][i];
            } 
            if (j != n)
            {
                a.push_back((edge){id(j, i), id(j + 1, i), dwn[j][i]});
                sufs[i] += dwn[j][i];
            }
        }
        solve(a, b, del);
        suf[i] = b;
        sufs[i] -= del;
    }
}

int main()
{
    gen();
    init_pre();
    init_suf();
    int l, r, i;
    read(q);
    for (i = 1; i <= n * m; i++) f[i] = i, bo[i] = 0;
    while (q--)
    {
        read(l); read(r);
        ans = pres[l - 1] + sufs[r + 1];
        vector<edge> a, b;
        for (i = 1; i <= n; i++)
        {
            ans += rht[i][m];
            a.push_back((edge){id(i, m), id(i, 1), rht[i][m]});
            upt(id(i, m), 0);
            upt(id(i, 1), 0);
            upt(id(i, l - 1), 0);
            upt(id(i, r + 1), 0);
        }
        int len = pre[l - 1].size();
        for (i = 0; i < len; i++) a.push_back(pre[l - 1][i]);
        len = suf[r + 1].size();
        for (i = 0; i < len; i++) a.push_back(suf[r + 1][i]);
        ll del = 0;
        solve(a, b, del);
        ans -= del;
        print(ans); putchar('\n');
    }
    return 0;
}

来源：CSDN

作者：花淇淋

链接：https://blog.csdn.net/qq_39694066/article/details/103823127

标签

MST

inline

「SDOI2019」世界地图

Address

Solution

Code