转载请注明出处,谢谢http://blog.csdn.net/ACM_cxlove?viewmode=contents    by---cxlove

上场CF的C题是一个树的分治。。。

今天刚好又看到一题,就做了下

题意:一棵树,问两个点的距离<=k的点对数目。

http://poj.org/problem?id=1987

貌似是经典的点分治题。。。。。

看成有根树,那么这样的点对路径分为两种,1、过根节点,2、存在于某一棵子树当中。

显然情况2可以看成是一种子情况

对于1的统计,统计所有节点到根节点的距离,枚举+二分可以得到有多少个二元组的和<=k。

但是需要除掉两个点都在某一棵子树中的情况,所以枚举所有子树,同样是枚举+二分。

至于根的选择,选取树的重心。。。我是两次DFS,类似数形DP,求出所有子树的size,找到某一个结点,若删除这个结点,剩下的子树的size中最大的最小。

总体复杂度大概是nlgnlgn

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>
#pragma comment(linker, "/STACK:1024000000,1024000000")
using namespace std;
const int N = 40005;
struct Edge{
int v,next,w;
}e[N<<1];
int tot,start[N],n,m,k,del[N],ans=0;
int size[N];
void _add(int u,int v,int w){
e[tot].v=v;e[tot].w=w;
e[tot].next=start[u];start[u]=tot++;
}
void add(int u,int v,int w){
_add(u,v,w);
_add(v,u,w);
}
void cal(int u,int pre){
size[u]=1;
for(int i=start[u];i!=-1;i=e[i].next){
int v=e[i].v;
if(v==pre||del[v]) continue;
cal(v,u);
size[u]+=size[v];
}
}
int newroot,maxsize,totalsize;
void dfs(int u,int pre){
int mx=0,sum=1;
for(int i=start[u];i!=-1;i=e[i].next){
int v=e[i].v;
if(v==pre||del[v]) continue;
dfs(v,u);
mx=max(mx,size[v]);
sum+=size[v];
}
mx=max(mx,totalsize-sum);
if(mx<maxsize){
maxsize=mx;
newroot=u;
}
}
int search(int r){
newroot=-1;maxsize=1<<30;
cal(r,-1);
totalsize=size[r];
dfs(r,-1);
return newroot;
}
int dist[N],idx;
vector<int>sub[N],all;
void gao(int u,int pre){
all.push_back(dist[u]);
sub[idx].push_back(dist[u]);
for(int i=start[u];i!=-1;i=e[i].next){
int v=e[i].v,w=e[i].w;
if(v==pre||del[v]) continue;
dist[v]=dist[u]+w;
gao(v,u);
}
}
void solve(int root){
root=search(root);
del[root]=1;
if(totalsize==1) return ;
idx=0;all.clear();
for(int i=start[root];i!=-1;i=e[i].next){
int v=e[i].v,w=e[i].w;
if(del[v]) continue;
sub[idx].clear();
dist[v]=w;
gao(v,-1);
sort(sub[idx].begin(),sub[idx].end());
idx++;
}
for(int i=0;i<idx;i++){
int pos;
for(int j=0;j<sub[i].size();j++){
pos=upper_bound(sub[i].begin(),sub[i].end(),k-sub[i][j])-sub[i].begin()-1;
if(pos>j) ans-=pos-j;
}
pos=upper_bound(sub[i].begin(),sub[i].end(),k)-sub[i].begin()-1;
if(pos>=0) ans+=pos+1;
}
sort(all.begin(),all.end());
for(int i=0;i<all.size();i++){
int pos=upper_bound(all.begin(),all.end(),k-all[i])-all.begin()-1;
if(pos>i) ans+=pos-i;
}
for(int i=start[root];i!=-1;i=e[i].next){
int v=e[i].v;
if(del[v]) continue;
solve(v);
}
}
int main(){
tot=0;memset(start,-1,sizeof(start));
scanf("%d%d",&n,&m);
for(int i=0;i<m;i++){
int u,v,w;char str[5];
scanf("%d%d%d%s",&u,&v,&w,str);
add(u,v,w);
}
scanf("%d",&k);
solve(1);
printf("%d\n",ans);
return 0;
}

POJ 1987 Distance Statistics(树的点分治)的更多相关文章

  1. POJ 1987 Distance Statistics 树分治

    Distance Statistics     Description Frustrated at the number of distance queries required to find a ...

  2. POJ 1987 Distance Statistics

    http://poj.org/problem?id=1987 题意:给一棵树,求树上有多少对节点满足距离<=K 思路:点分治,我们考虑把每个距离都存起来,然后排序,一遍扫描计算一下,注意还要减掉 ...

  3. poj 1741 Tree(树的点分治)

    poj 1741 Tree(树的点分治) 给出一个n个结点的树和一个整数k,问有多少个距离不超过k的点对. 首先对于一个树中的点对,要么经过根结点,要么不经过.所以我们可以把经过根节点的符合点对统计出 ...

  4. POJ 1987 BZOJ 3365 Distance Statistics 树的分治(点分治)

    题目大意:(同poj1741,刷一赠一系列) CODE: #include <cstdio> #include <cstring> #include <iostream& ...

  5. 【POJ 1741】 Tree (树的点分治)

    Tree   Description Give a tree with n vertices,each edge has a length(positive integer less than 100 ...

  6. POJ 1741 Tree (树的点分治入门)

      Tree Time Limit: 1000MS   Memory Limit: 30000K Total Submissions: 16172   Accepted: 5272 Descripti ...

  7. POJ 2255 Tree Recovery 树的遍历,分治 难度:0

    http://poj.org/problem?id=2255 #include<cstdio> #include <cstring> using namespace std; ...

  8. POJ 1741/1987 树的点分治

    树的点分治,主要思想是每次找子树的重心,计算经过根节点的情况数,再减去点对属于同一子树的情况. #include <iostream> #include <vector> #i ...

  9. poj 1741 树的点分治(入门)

    Tree Time Limit: 1000MS   Memory Limit: 30000K Total Submissions: 18205   Accepted: 5951 Description ...

随机推荐

  1. [转]laravel 4之视图及Responses

    http://dingjiannan.com/2013/laravel-responses/   laravel 4之视图及Responses 16 Aug 2013 Laravel的Response ...

  2. 一个用 Cumulative Penalty 培训 L1 正规 Log-linear 型号随机梯度下降

      Log-Linear 模型(也叫做最大熵模型)是 NLP 领域中使用最为广泛的模型之中的一个.其训练常採用最大似然准则.且为防止过拟合,往往在目标函数中增加(能够产生稀疏性的) L1 正则.但对于 ...

  3. css_day5

  4. IoC容器Autofac正篇之解析获取(五)

    解析获取的方式有如下几种: Resolve class Program { static void Main(string[] args) { var builder = new ContainerB ...

  5. bower安装使用以及git安装

    bower需要:node 和 git node安装包下载:http://blog.csdn.net/myan/article/details/2028545 Git安装: 选择第二项:Use Git ...

  6. 已知TSP问题的最好解

    a280 : 2579ali535 : 202339att48 : 10628att532 : 27686bayg29 : 1610bays29 : 2020berlin52 : 7542bier12 ...

  7. (转) int argc, char* argv[] 的用法

    int main(int argc, char* argv[]) 這兩個參數的作用是什麼呢?argc 是指命令行輸入參數的個數,argv存儲了所有的命令行參數.假如你的程式是hello.exe,如果在 ...

  8. mysql导入.sql文件

    1. source /home/susie ...../**.sql 2. \. /home/susie/.../**.sql 批量导入.sql文件 首先新建一个main.sql,然后在main.sq ...

  9. ccf练习---节日

    问题描述 有一类节日的日期并不是固定的,而是以“a月的第b个星期c”的形式定下来的,比如说母亲节就定为每年的五月的第二个星期日. 现在,给你a,b,c和y1, y2(1850 ≤ y1, y2 ≤ 2 ...

  10. NET Core开发-获取所有注入(DI)服务

    NET Core开发-获取所有注入(DI)服务 获取ASP.NET Core中所有注入(DI)服务,在ASP.NET Core中加入了Dependency Injection依赖注入. 我们在Cont ...