【GDOI2017第二轮模拟day1】公路建设(克鲁斯卡尔最小生成树+线段树+归并)

题目大意:

给出n个点,m条边。
q组询问,每次询问编号在[l..r]范围的所有边做克鲁斯卡尔最小生成树的代价。

1<=n<=100,1<=m<=100000,1<=q<=15000

题解:

n好小啊。

可不可以线段树啊。

套个归并就可以啊。

莫队+lct可不可以啊。

时间复杂度:O(αmn+α q log m n)

Code:

#include<cstdio>
#include<cstring>
#define fo(i, x, y) for(int i = x; i <= y; i ++)
using namespace std;

const int N = 1e5 + 5;

int n, m, q;
struct edge {
    int u, v, c;
} e[N];
int f[N];

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

struct tree {
    int s, d[100];
} t[N * 4];

void He(tree &t, tree a, tree b) {
    t.s = t.d[0] = 0;
    fo(i, 1, n) f[i] = i;
    int l = 1, r = 1;
    while(l <= a.d[0] && r <= b.d[0]) {
        if(e[a.d[l]].c < e[b.d[r]].c) {
            if(find(e[a.d[l]].u) != find(e[a.d[l]].v)) {
                t.s += e[a.d[l]].c;
                t.d[++ t.d[0]] = a.d[l];
                f[f[e[a.d[l]].u]] = f[e[a.d[l]].v];
            } 
            l ++;
        } else {
            if(find(e[b.d[r]].u) != find(e[b.d[r]].v)) {
                t.s += e[b.d[r]].c;
                t.d[++ t.d[0]] = b.d[r];
                f[f[e[b.d[r]].u]] = f[e[b.d[r]].v];
            } 
            r ++;
        }
    }
    for(; l <= a.d[0]; l ++) {
        if(find(e[a.d[l]].u) != find(e[a.d[l]].v)) {
            t.s += e[a.d[l]].c;
            t.d[++ t.d[0]] = a.d[l];
            f[f[e[a.d[l]].u]] = f[e[a.d[l]].v];
        } 
    }
    for(; r <= b.d[0]; r ++) {
        if(find(e[b.d[r]].u) != find(e[b.d[r]].v)) {
            t.s += e[b.d[r]].c;
            t.d[++ t.d[0]] = b.d[r];
            f[f[e[b.d[r]].u]] = f[e[b.d[r]].v];
        } 
    }
}

void Build(int i, int x, int y) {
    if(x == y) {
        if(e[x].u != e[x].v) {
            t[i].d[0] = 1; t[i].d[1] = x;
            t[i].s = e[x].c;
        }
        return;
    }
    int m = x + y >> 1;
    Build(i + i, x, m); Build(i + i + 1, m + 1, y);
    He(t[i], t[i + i], t[i + i + 1]);
}

tree p;

tree find(int i, int x, int y, int l, int r) {
    if(x == l && y == r) return t[i];
    int m = x + y >> 1;
    if(r <= m) return find(i + i, x, m, l, r);
    if(l > m) return find(i + i + 1, m + 1, y, l, r);
    He(p, find(i + i, x, m, l, m), find(i + i + 1, m + 1, y, m + 1, r));
    return p;
}

int x, y;

int main() {
    freopen("highway.in", "r", stdin);
    freopen("highway.out", "w", stdout);
    scanf("%d %d %d", &n, &m, &q);
    fo(i, 1, m) scanf("%d %d %d", &e[i].u, &e[i].v, &e[i].c);
    Build(1, 1, m);
    for(; q; q --) {
        scanf("%d %d", &x, &y);
        p = find(1, 1, m, x, y);
        printf("%d\n", p.s);
    }
}
  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值