【bzoj2435】道路修建

今天真是愉♂快的一天……
QAQQQQQQQ

大背景：我想学主席树

---

想学主席树，有人说你得先学可持久化

---

然后就去学可持久化
我找到WJMZBMR，他说：“就凭你也敢看我课件？？？”
然后我就去找fhq，fhq亲切友好的与我交谈了半天，然后他说想学这个先去学LCT

---

然后我就跑去学LCT
我找到了DQS
DQS说：“你还不会链剖你学个篮子”

---

然后我就去学链剖
结果fhq说：“你等等，我给你个水题……”
我回头他就给我这个题……
嗯

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <cmath>
using namespace std;
typedef long long LL;
const int MAXN = 1000000 + 5;
struct edge
{
    int f,t,v;
}l[MAXN << 1];
int n,m;
int first[MAXN],next[MAXN << 1],tot;
int deep[MAXN],sz[MAXN];
bool use[MAXN];
void init()
{
    memset(first,0xfff,sizeof(first));
    memset(deep,0,sizeof(deep));
    memset(use,0,sizeof(use));
    memset(sz,0,sizeof(sz));
    tot = 0;
    return;
}
void build(int f,int t,int v)
{
    l[++tot] = (edge){f,t,v};
    next[tot] = first[f];
    first[f] = tot;
    return;
}
void dfs(int x,int f)
{
    deep[x] = deep[f] + 1;
    use[x] = true;
    sz[x] = 1;
    for(int i = first[x];i != -1;i = next[i])
    {
        int v = l[i].t;
        if(use[v])
            continue;
        dfs(v,x);
        sz[x] += sz[v];
    }
    return;
}
LL solve()
{
    dfs(1,0);
    LL ans = 0;
    for(int i = 1;i <= tot;i += 2)
    {
        int u = l[i].f,v = l[i].t;
        if(deep[u] > deep[v])
            swap(u,v);
        ans += (LL)((LL)l[i].v * abs((LL)sz[v] - (LL)(n - sz[v])));
    }
    return ans;
}
int f,t,v;
int main()
{
    init();
    scanf("%d",&n);
    for(int i = 1;i < n;i ++)
    {
        scanf("%d %d %d",&f,&t,&v);
        build(f,t,v);
        build(t,f,v);
    }
    printf("%lld\n",solve());
    return 0;
}

转载于:https://my.oschina.net/u/2992707/blog/778038