[JZOJ5060] 公路建设

题目描述

　　在Byteland一共有n个城市，编号依次为1到n，它们之间计划修建m条双向道路，其中修建第i条道路的费用为ci。

　　Byteasar作为Byteland公路建设项目的总工程师，他决定选定一个区间[l,r]，仅使用编号在该区间内的道路。他希望选择一些道路去修建，使得连通块的个数尽量少，同时，他不喜欢修建多余的道路，因此每个连通块都可以看成一棵树的结构。

　　为了选出最佳的区间，Byteasar会不断选择q个区间，请写一个程序，帮助Byteasar计算每个区间内修建公路的最小总费用。

数据分析

　　

分析

　　为什么数组开小了是WA不是RE，害我调了一个晚上啊

　　

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>
#include <queue>
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define N 105
#define M 100005
#define lc (p << 1)
#define rc (p << 1 | 1)
#define mid (l + r >> 1)

int n, m, q;
int f[N], tmp[N << 1];

struct Edge {int u, v, w;} edge[M];
struct Tree {int sum, len, e[N];} t[M << 2];

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

Tree merge(Tree l, Tree r) {
    Tree p; p.sum = p.len = 0;
    int i = 1, j = 1, k = 0;
    while (i <= l.len && j <= r.len) {
        if (edge[l.e[i]].w < edge[r.e[j]].w)
            tmp[++k] = l.e[i++];
        else tmp[++k] = r.e[j++];
    }
    while (i <= l.len) tmp[++k] = l.e[i++];
    while (j <= r.len) tmp[++k] = r.e[j++];
    for (int i = 1; i <= n; i++) f[i] = i;
    for (int i = 1; i <= k; i++) {
        int x = find(edge[tmp[i]].u);
        int y = find(edge[tmp[i]].v);
        if (x == y) continue;
        f[x] = y;
        p.e[++p.len] = tmp[i];
        p.sum += edge[tmp[i]].w;
    }
    return p;
}

void build(int p, int l, int r) {
    if (l == r) {
        t[p].e[t[p].len = 1] = l; t[p].sum = edge[l].w;
        return;
    }
    build(lc, l, mid);
    build(rc, mid + 1, r);
    t[p] = merge(t[lc], t[rc]);
}

Tree query(int p, int l, int r, int ql, int qr) {
    if (l == ql && r == qr) return t[p];
    if (qr <= mid) return query(lc, l, mid, ql, qr);
    if (ql > mid) return query(rc, mid + 1, r, ql, qr);
    return merge(query(lc, l, mid, ql, mid), query(rc, mid + 1, r, mid + 1, qr));
}

int main() {
    scanf("%d%d%d", &n, &m, &q);
    for (int i = 1; i <= m; i++)
        scanf("%d%d%d", &edge[i].u, &edge[i].v, &edge[i].w);
    build(1, 1, m);
    while (q--) {
        int l, r;
        scanf("%d%d", &l, &r);
        printf("%d\n", query(1, 1, m, l, r).sum);
    }
    
    return 0;
}

View Code

标签：int,公路,mid,Byteasar,建设,JZOJ5060,query,include,define
来源： https://www.cnblogs.com/Pedesis/p/11354901.html