题目链接
题意:
有n个点,编号1-n
保证这n个点来自于一个树,给出n-1条边的连接方式
问有多少种对(u,v) u<v满足u v之间的路上 没有长度超过q的边
样例+题目解释:
例如第一组样例构成的树如图所示
询问分别为:
5 2 3 4 1
第一次询问:
(1,2)(1,3)(1,4)(1,5)(1,6)(1,7)
(2,3)(2,4)(2,5)(2,6)(2,7)
(3,4)(3,5)(3,6)(3,7)
(4,5)(4,6)(4,7)
(5,6)(5,7)
(6,7)
共21条路径满足条件
第二次询问:
(1,2)(1,4)(1,5)
(2,4)(2,5)
(3,6)
(4,5)
共7条路径满足条件
……(省略其他询问)
思路:
刚开始看到这道题发现难度居然还没有1213F的难度高 和1213E难度一样,就果断来先补这道题了。
结果一看题发现和E完全不是一个难度水平啊…可能我这方面比较菜…
然后去看了别人AC的代码,还让队友给讲了才会写(我太菜了)
就来好好补这道题了。。
题目中说了这n个点来自于一个树,所以不用考虑某两个点之间不相连的情况
首先把这n-1条边按weight从小到大排序
然后利用并查集找出第i条边满足条件时,总共的情况数ans[i]
还用第一组样例来解释吧:
把n条边按weight排完序以后为:
1 2 1
2 4 1
4 5 2
3 6 2
3 2 3
5 7 4
然后用getans函数来得到每个ans的值
getans函数:(这里是我当初最难看懂的…)
生成第一条边时,将1的根节点变成2,2的sum值变成2,代表这一堆里现在有两个点
(把2的根节点变成1也是没问题的,但是要注意是把1的sum值变成2,因为之后每次要用的都是根节点的sum值,注意要对应)
ans[1]=0+11
//ans[i]表示第i条边满足条件时的情况总数
因为边是按weight从小到大排好的,所以满足ans[i-1]的边肯定满足ans[i]
11的意思在下面稍微复杂一点的地方解释
生成第二条边时,新组成的堆里的 新组成的边的个数为sum[2]*sum[1],还需要再加上之前满足的ans[i-1]条边,同样和上面一样,把根节点变成4
……(省略之后的生成)
q次询问之后要用二分查找答案
因为边都是按从小到大排好的,所以只需找到满足 k[i].weight≤q的i的最大即可,ans[i]即为对应的答案。
感觉写完博客以后自己是真的懂了。。。如果还有哪里不小心写错的话,欢迎指出hhh
代码:
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll maxn=200005;
ll n,m,sum[maxn],f[maxn],ans[maxn],i,j;
struct hh
{
ll a,b,w;
} k[maxn];
bool cmp(hh k1,hh k2)
{
return k1.w<k2.w;
}
ll fin(ll x)
{
if(f[x]==x)
return x;
return f[x]=fin(f[x]);
}
void getans(ll a,ll b)
{
a=fin(a);
b=fin(b);
ans[i]=ans[i-1]+sum[a]*sum[b];
sum[b]+=sum[a];
f[a]=b;
}
int main()
{
scanf("%lld%lld",&n,&m);
for(i=1; i<=n-1; i++)
{
scanf("%lld%lld%lld",&k[i].a,&k[i].b,&k[i].w);
f[i]=i;
sum[i]=1;
}
f[n]=n,sum[n]=1;
sort(k+1,k+n,cmp);
for(i=1; i<=n-1; i++)
{
getans(k[i].a,k[i].b);
}
while(m--)
{
ll z;
scanf("%lld",&z);
ll l=0,r=n-1,mid,ans1=0;
while(l<=r)
{
mid=(l+r)/2;
if(k[mid].w<=z)
{
l=mid+1;
ans1=max(ans1,mid);
}
else
r=mid-1;
}
printf("%lld ",ans[ans1]);
}
}