poj-1741（树分治点权）

题目链接

思路：

//#pragma comment(linker, "/STACK:1024000000,1024000000")
#include<iostream>
#include<stdio.h>
#include<math.h>
#include <string>
#include<string.h>
#include<map>
#include<queue>
#include<set>
#include<utility>
#include<vector>
#include<algorithm>
#include<stdlib.h>
using namespace std;
#define eps 1e-8
#define pii pair<int,int>
#define inf 0x3f3f3f3f
#define rd(x) scanf("%d",&x)
#define rd2(x,y) scanf("%d%d",&x,&y)
#define ll long long int
#define mod 1000000007
#define maxn 11000
#define maxm 10001005
int mi(int a,int b){return a<b?a:b;}
int ma(int a,int b){return a>b?a:b;}
int n,k,mm,root;
int ans;
int vis[maxn],sz[maxn],mx[maxn];
int dis[maxn],num;
vector< pair<int,int> > vv[maxn];
void dfsize(int x,int fa){//统计子树大小sz[x]=1;mx[x]=0;for(int i=0;i<vv[x].size();i++){pii p=vv[x][i];int v=p.first;if(v==fa||vis[v]) continue;dfsize(v,x);sz[x]+=sz[v];if(sz[v]>mx[x]) mx[x]=sz[v];}
}
void dfsroot(int r,int x,int fa){//求重心if(sz[r]-sz[x]>mx[x]) mx[x]=sz[r]-sz[x];if(mx[x]<mm) mm=mx[x],root=x;for(int i=0;i<vv[x].size();i++){pii p=vv[x][i];int v=p.first;if(v==fa||vis[v]) continue;dfsroot(r,v,x);}
}
void dfsdis(int u,int d,int fa){//求距离dis[num++]=d;for(int i=0;i<vv[u].size();i++){pii p=vv[u][i];int v=p.first;if(fa==v||vis[v]) continue;dfsdis(v,d+p.second,u);}
}
int calc(int u,int d){//统计经过u的路径个数和，其中包括在同一子树路径上的int res=0;num=0;dfsdis(u,d,0);sort(dis,dis+num);int i=0,j=num-1;while(i<j){while(dis[i]+dis[j]>k&&i<j) j--;res+=j-i;i++;}return res;
}
void dfs(int u){mm=n;dfsize(u,0);// root=u;dfsroot(u,u,0);ans+=calc(root,0);vis[root]=1;int rt=root;for(int i=0;i<vv[rt].size();i++){pii p=vv[rt][i];int v=p.first;if(vis[v]) continue;ans-=calc(v,p.second);//减去belongi==belongj的个数dfs(v);}
}
int main()
{int u,v,z;while(rd2(n,k)&&(n||k)){for(int i=0;i<=n;i++) vv[i].clear();for(int i=1;i<n;i++){scanf("%d%d%d",&u,&v,&z);vv[u].push_back(make_pair(v,z));vv[v].push_back(make_pair(u,z));}memset(vis,0,sizeof(vis));ans=0;dfs(1);printf("%d\n",ans);}return 0;
}

poj-1741（树分治点权）相关推荐

POJ 1741 树分治
题目链接[http://poj.org/problem?id=1741] 题意: 给出一颗树,然后寻找点对(u,v)&&dis[u][v] < k的对数. 题解: 这是一个很经典 ...
POJ 1741 Tree 树分治
题意: 给出一颗有$n (n \leq 10^4)$个节点的树,和一个$k$.统计有多少个点对$u, \, v(u \neq v)$满足$u$到$v$的最短距离不超过$k$. ...
Tree（树分治入门）
题目链接:http://poj.org/problem?id=1741 Tree Time Limit: 1000MS Memory Limit: 30000K Total Submissions ...
POJ1741：Tree——题解+树分治简要讲解
http://poj.org/problem?id=1741 题目大意:给一棵树,求点对间距离<=k的个数. -------------------- 以这道题为例记录一下对于树分治的理解. 树 ...
[POJ 1741] Tree
Link: POJ 1741 传送门 Solution: 此题的难点在于点分治上的统计如果依然采取每棵子树的结果与之前所有子树的结果合并的方式会比较麻烦同时复杂度可能超过$O(n*log(n))$ ...
POJ1741 Tree 树中点对统计【树分治入门】
算法合集之<分治算法在树的路径问题中的应用> 论文下载地址树被定义为没有圈的连通图,有几个性质在树中去掉一条边后,得到的图是不连通的在树中添加一条边后,一定存在一条边树的每一对顶点 ...
树分治树链剖分相关题目讨论
预备知识树分治,树链剖分 poj1741 •一棵有n个节点的树,节点之间的边有长度.方方方想知道,有多少个点对距离不超过m 题解点分治模板题.详见我早上写的http://www.cnblogs.c ...
BZOJ 2566 xmastree（树分治+multiset）
题目链接:http://www.lydsy.com:808/JudgeOnline/problem.php?id=2566 题意:一棵有边权的树.结点有颜色.每次修改一个点的颜色.求每次修改后所有同色 ...
P4719 【模板】“动态 DP“动态树分治（矩阵/轻重链剖分/ddp）
P4719 [模板]"动态 DP"&动态树分治求解树上最大权独立集,但是需要支持修改. https://www.luogu.com.cn/problem/solution ...

poj-1741（树分治点权）

poj-1741（树分治点权）相关推荐

最新文章

热门文章

poj-1741（树分治 点权）

poj-1741（树分治 点权）相关推荐

最新文章

热门文章

poj-1741（树分治点权）

poj-1741（树分治点权）相关推荐