问题
http://hihocoder.com/contest/hiho112/problem/1
问题分析
题目给定一课最小生成树,求所有节点对之间的距离,然后将所有距离相加之后的和。而且之后要修改一些边的权重,再查询距离。
很容易想到一个方法是:统计每条边会被用到几次,最终答案就是 每条边的权重*用到的次数 然后所有边求和。 由于树的形状是不变的,因此每条边被用到的次数是可以复用。每次修改边的长度后,只需要对被修改的边重新计算,然后更新答案。
而要统计每条边被用几次,可以看做去除当前边后树被分成两部分,两部分节点个数的乘积即为此边被使用的次数。
如果我们使用dfs 遍历整棵树,可以求出每条边所指向的子树对应的节点个数,另外一部分节点的个数即为总节点数- 子树节点数。
另外为了在修改边时方便,我们需要求出每个节点所在的层数, 将子树的节点个数存储在子树的根节点处。当修改时只需要找出要修改的边所对应的子树的根节点。
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
enum{maxn= 100006};
vector<int> tree[maxn];
vector<int> treeLen[maxn];
ll total;
int level[maxn];
int nodeNum[maxn];
int path[maxn];
int N, M;
void dfs(int f)
{
nodeNum[f] = 1;
for (int i=0; i<tree[f].size(); ++i)
{
int v = tree[f][i];
if (level[v] == 0)
{
path[v] = treeLen[f][i]; // 边的长度存储在子树的根节点处。
level[v] = level[f]+1;
dfs(v);
nodeNum[f] += nodeNum[v];
total += (ll)(N-nodeNum[v]) * (ll)nodeNum[v] * (ll)path[v];
}
}
}
void edit(int a, int b, int c)
{
if (level[a] < level[b])
swap(a, b);
int delt = c - path[a];
path[a] = c;
total += (ll)delt * (ll)(N-nodeNum[a]) * (ll)nodeNum[a];
}
int main()
{
scanf("%d %d", &N, &M);
for (int i=0; i<N-1; ++i)
{
int a, b, c;
scanf("%d %d %d", &a, &b, &c);
tree[a].push_back(b);
treeLen[a].push_back(c);
tree[b].push_back(a);
treeLen[b].push_back(c);
}
total =0;
memset(level, 0, sizeof(level));
level[1] = 1;
dfs(1);
char q[100];
for (int i=0; i<M; ++i)
{
scanf("%s", q);
if (q[0] == 'Q')
{
printf("%lld\n", total);
}else
{
int a, b, c;
scanf("%d %d %d", &a, &b, &c);
edit(a, b, c);
}
}
return 0;
}
注意问题
结果比较大需要用 long long 存储, 答案计算过程中也要转换为long long类型;