HDU 4729 An Easy Problem for Elfness 主席树

题意：

给出一棵树，每条边有一个容量。
有若干次询问：\(S \, T \, K \, A \, B\)，求路径\(S \to T\)的最大流量。
有两种方法可以增大流量：

• 花费\(A\)可以新修一条管道，管道可以连接任意两个点，两个点之间可以有任意条管道连接，新修的管道容量为\(1\)
• 花费\(B\)可以使某条管道(包括新修的管道)的容量增加\(1\)

求在不超过预算\(K\)的情况下的最大流量。

分析：

注意到流量最大只有\(10^4\)，所以我们可以直接在值域上建一棵主席树。
主席树区间维护的是管道的个数以及它们的容量之和，这样方便后面的计算。
首先我们可以求一下路径上的第\(1\)小的容量，这是初始容量。
按照\(A, B\)的大小关系，有两种情况：

• \(A \leq B\)，把所有的经费都用来修建新管道上。因为扩容一次不一定能使流量增加，但是新修管道一定可以增加流量，而且还便宜。
• \(A > B\)，这样就不能随便新建管道了，所以有两种可能：
  • 只新建一条管道，然后再这条新管道上扩容
  • 只在原来的管道上扩容

在求能扩容得到的最大容量可以用二分的思想，往主席树的左右子树走下去。
另：根据题面的数据范围，最终答案可能会爆int的，但是事实证明测试数据中没有这样的数据。。

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

const int maxn = 100000 + 10;
const int M = 20;
int n, m;

struct Edge
{
    int v, w, nxt;
    Edge() {}
    Edge(int v, int w, int nxt): v(v), w(w), nxt(nxt) {}
};

int ecnt, head[maxn];
Edge edges[maxn * 2];

void AddEdge(int u, int v, int w) {
    edges[ecnt] = Edge(v, w, head[u]);
    head[u] = ecnt++;
}

struct Node
{
    int lch, rch, cnt, sum;
}T[maxn << 5];
int sz, root[maxn];

int update(int pre, int L, int R, int pos) {
    int rt = ++sz;
    T[rt] = T[pre];
    T[rt].cnt++;
    T[rt].sum += pos;
    if(L < R) {
        int M = (L + R) / 2;
        if(pos <= M) T[rt].lch = update(T[pre].lch, L, M, pos);
        else T[rt].rch = update(T[pre].rch, M+1, R, pos);
    }
    return rt;
}

int query(int u, int v, int l, int L, int R, int k) {
    if(L == R) return L;
    int M = (L + R) / 2;
    int sum = T[T[u].lch].cnt + T[T[v].lch].cnt - T[T[l].lch].cnt * 2;
    if(sum >= k) return query(T[u].lch, T[v].lch, T[l].lch, L, M, k);
    else return query(T[u].rch, T[v].rch, T[l].rch, M+1, R, k - sum);
}

int fa[maxn], dep[maxn];

void dfs(int u) {
    for(int i = head[u]; ~i; i = edges[i].nxt) {
        int v = edges[i].v;
        if(v == fa[u]) continue;
        fa[v] = u;
        dep[v] = dep[u] + 1;
        root[v] = update(root[u], 0, M, edges[i].w);
        dfs(v);
    }
}

int anc[maxn][20];

void preprocess() {
    memset(anc, 0, sizeof(anc));
    for(int i = 1; i <= n; i++) anc[i][0] = fa[i];
    for(int j = 1; (1 << j) < n; j++)
        for(int i = 1; i <= n; i++) if(anc[i][j-1])
            anc[i][j] = anc[anc[i][j-1]][j-1];
}

int LCA(int u, int v) {
    if(dep[u] < dep[v]) swap(u, v);
    int log;
    for(log = 0; (1 << log) < dep[u]; log++);
    for(int i = log; i >= 0; i--)
        if(dep[u] - (1<<i) >= dep[v]) u = anc[u][i];
    if(u == v) return u;
    for(int i = log; i >= 0; i--)
        if(anc[u][i] && anc[u][i] != anc[v][i])
            u = anc[u][i], v = anc[v][i];
    return fa[u];
}

int bsearch(int u, int v, int l, int x) {
    int L = 0, R = M, cnt = 0, sum = 0;
    while(L < R) {
        int mid = (L + R) / 2;
        int tcnt = T[T[u].lch].cnt + T[T[v].lch].cnt - T[T[l].lch].cnt * 2;
        int tsum = T[T[u].lch].sum + T[T[v].lch].sum - T[T[l].lch].sum * 2;
        if((cnt + tcnt) * mid - tsum - sum > x) {
            u = T[u].lch, v = T[v].lch, l = T[l].lch;
            R = mid;
        } else {
            cnt += tcnt; sum += tsum;
            u = T[u].rch, v = T[v].rch, l = T[l].rch;
            L = mid + 1;
        }
    }
    return L - 1;
}

int main()
{
    int _; scanf("%d", &_);
    for(int kase = 1; kase <= _; kase++) {
        printf("Case #%d:\n", kase);

        scanf("%d%d", &n, &m); sz = 0;
        ecnt = 0;
        memset(head, -1, sizeof(head));
        for(int i = 1; i < n; i++) {
            int u, v, w; scanf("%d%d%d", &u, &v, &w);
            AddEdge(u, v, w); AddEdge(v, u, w);
        }

        dfs(1); preprocess();
        while(m--) {
            int u, v, k, a, b;
            scanf("%d%d%d%d%d", &u, &v, &k, &a, &b);
            int lca = LCA(u, v);
            int base = query(root[u], root[v], root[lca], 0, M, 1);
            if(a <= b) printf("%d\n", base + k / a);
            else {
                int ans = base;
                if(k >= a) ans += (k - a) / b + 1;
                int x = k / b;
                ans = max(ans, bsearch(root[u], root[v], root[lca], x));
                printf("%d\n", ans);
            }
        }
    }

    return 0;
}

转载于:https://www.cnblogs.com/AOQNRMGYXLMV/p/5295467.html