codeforces 500D New Year Santa Network (树结构, 组合, 概率(注意精度))

17 篇文章 0 订阅
12 篇文章 0 订阅

题意:给了一颗树,和它的边权。。然后给若干修改,每次改一条边的权值。。问题是,每次修改后随机找三个不同的点,求 E( d(a, b)+d(a, c)+d(b, c)) 。。。

思路:

如果是随机选两个点,就好办得多了。

我们让1作为root, 如果 (u, v)  修改后变化了 d, 那么它对总权值的贡献是 size(k)*(n-size(k))*d, k = size(u) < size(v) ? u : v

size(x) 是 以x为根的子树中的节点数, 这里有两个集合,u的子树, 除了u的子树外其它所有点,从这两个集合中各选一个点,其路径必然包含(u, v)

--------------------------------

如果是选三个点呢, 先展开。。

写成二元组的和后,惊奇的发现 = = 边权和刚好是 任选两点得到的路径权值和的 n-2 倍。。。

想一想,(a, b) 可以和另外 n-2 个点构成三元组,所以有这个结果。。。

所以,我们先dfs一次,求出每个节点的size, 然后维护边权和就行了。

#include<bits/stdc++.h>
using namespace std;

#define SPEED_UP iostream::sync_with_stdio(false);
#define FIXED_FLOAT cout.setf(ios::fixed, ios::floatfield);
#define rep(i, s, t) for(int (i)=(s);(i)<=(t);++(i))
#define urep(i, s, t) for(int (i)=(s);(i)>=(t);--(i))

typedef long long LL;

const int Maxn = ((int)1e5) + 100;

struct Edge{
    int u, v;
    LL w;
    Edge(){}
    Edge(int u, int v, int w):u(u), v(v), w(w){}
    int getAdj(int x) {
        return x == u ? v : u;
    }
};
Edge E[Maxn];
vector<int> g[Maxn];
LL n, num[Maxn], tot;
double base;

LL dfs(int now, int fa) {
    num[now] = 1;
    int sz = g[now].size();
    rep(i, 0, sz-1) {
        Edge& e = E[g[now][i]];
        int u = e.getAdj(now);
        if ( u != fa) {
            num[now] += dfs(u, now);
            tot += e.w * 1ll * num[u] * (n - num[u]);
        }
    }
    //cout << "node " << now << " size: " << num[now] << endl;
    return num[now];
}

void init() {
    tot = 0;
    base = 6.0 / (n*(n-1));
    memset(num, 0, sizeof(num));
    dfs(1, -1);
}

double update(int e_idx, LL w) {
    Edge & e = E[e_idx];
    LL d = w - e.w;
    LL t = num[e.u];
    if (num[e.u] > num[e.v]) t = num[e.v];

    tot += t * (n-t) * d;
    //cout << "update " << e_idx << " old: " << E[e_idx].w << " new: " << w << endl;
    e.w = w;
    return tot * base;
}

int main() {
#ifndef ONLINE_JUDGE
    freopen("input.in", "r", stdin);
#endif
    SPEED_UP
    cin >> n;
    rep(i, 1, n-1) {
        int a, b, c;
        cin >> a >> b >> c;
        E[i] = Edge(a, b, c);
        g[a].push_back(i);
        g[b].push_back(i);
    }

    init();
/*
    cout << "tot: " << tot << ' ' << "real tot: " << tot*(n-2) << endl;
    cout << "base: " << base << endl;
    rep(i, 1, n) {cout << "num " << i << ": " << num[i] << endl;}
*/
    int q;
    cin >> q;
    FIXED_FLOAT
    rep(i, 1, q) {
        int r;
        LL w;
        cin >> r >> w;
        cout << std::setprecision(10) << update(r, w) << endl;
    }

    return 0;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值