【POJ1987】Distance Statistics ==【POJ1741】 树分治

广告:

#include <stdio.h>
int main()
{
    puts("转载请注明出处[vmurder]谢谢");
    puts("网址:blog.csdn.net/vmurder/article/details/44307489");
}

题意&题解

http://blog.csdn.net/vmurder/article/details/44302921

代码:(同一道题)

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define N 50100
#define inf 0x3f3f3f3f
using namespace std;
int n,m;
struct Eli
{
    int v,len,next;
}e[N<<1];
int head[N],cnt;
bool rem[N];
inline void add(int u,int v,int len)
{
    e[++cnt].v=v;
    e[cnt].len=len;
    e[cnt].next=head[u];
    head[u]=cnt;
}
int f[N],f2[N],son[N];
int TreeCenter,length;
int dfs1(int x,int p)
{
    int i,v;
    f[x]=f2[x]=0;
    for(i=head[x];i;i=e[i].next)
    {
        v=e[i].v;
        if(v==p||rem[v])continue;
        int temp=dfs1(v,x);
        if(temp>f[x])
        {
            f2[x]=f[x];
            f[x]=temp;
            son[x]=v;
        }
        else if(temp>f2[x])
            f2[x]=temp;
    }
    return f[x]+1;
}
int g[N];
void dfs2(int x,int p)
{
    int i,v;
    if(max(f[x],g[x])<length)
    {
        length=max(f[x],g[x]);
        TreeCenter=x;
    }
    for(i=head[x];i;i=e[i].next)
    {
        v=e[i].v;
        if(v==p||rem[v])continue;
        if(v==son[x])g[v]=max(g[x],f2[x])+1;
        else g[v]=max(g[x],f[x])+1;
        dfs2(v,x);
    }
    return ;
}
inline int get_TreeCenter(int x)
{
    if(rem[x])return 0;
    length=inf,g[x]=0;
    dfs1(x,0),dfs2(x,0);
    return TreeCenter;
}
struct Summorer
{
    int a,b;
    // 权值,子树标号
    Summorer(int _a=0,int _b=0):a(_a),b(_b){}
}src[N];
inline int cmpa(Summorer a,Summorer b)
{
    return a.a<b.a;
}
inline int cmpb(Summorer a,Summorer b)
{
    return a.b==b.b?a.a<b.a:a.b<b.b;
}
int size[N];
void dfs3(int x,int p,int t,int len)
{
    src[++cnt]=Summorer(len,t);
    size[x]=1;
    int i,v;
    for(i=head[x];i;i=e[i].next)
    {
        v=e[i].v;
        if(rem[v]||v==p)continue;
        dfs3(v,x,t,len+e[i].len);
        size[x]+=size[v];
    }
    return ;
}
int ans;
void work(int x,int n)
{
    if(!n)return ;
    rem[x=get_TreeCenter(x)]=1;
    int num=0,i,j,k,v,p;
    cnt=0;
    for(i=head[x];i;i=e[i].next)
    {
        v=e[i].v;
        if(rem[v])continue;
        dfs3(v,x,++num,e[i].len);
    }
    sort(src+1,src+cnt+1,cmpa);
    p=n;
    for(i=0;i<=n;i++)
    {
        while(i<p&&src[i].a+src[p].a>m)p--;
        if(i>=p)break;
        ans+=(p-i);
    }
    sort(src+1,src+cnt+1,cmpb);
    src[cnt+1].b=-1;
    for(i=0;i<=n;i=j)
    {
        for(j=i+1;src[j].b==src[i].b;j++);
        for(p=j-1,k=i;k<j;k++)
        {
            while(k<p&&src[k].a+src[p].a>m)p--;
            if(k>=p)break;
            ans-=(p-k);
        }
    }
    for(i=head[x];i;i=e[i].next)
    {
        v=e[i].v;
        if(rem[v])continue;
        work(v,size[v]-1);
    }
    return ;
}
void init()
{
    ans=cnt=0;
    memset(rem,0,sizeof rem);
    memset(head,0,sizeof head);
}
char useless[5];
int main()
{
    freopen("test.in","r",stdin);
    int i,a,b,c;

    while(scanf("%d%d",&n,&m)!=EOF)
    {
        init();
        while(m--)
        {
            scanf("%d%d%d%s",&a,&b,&c,useless);
            add(a,b,c),add(b,a,c);
        }
        scanf("%d",&m);
        work(1,n-1);
        printf("%d\n",ans);
    }

    return 0;
}
时间: 2024-11-08 22:37:44

【POJ1987】Distance Statistics ==【POJ1741】 树分治的相关文章

POJ1987——Distance Statistics

Distance Statistics Time Limit: 2000MS   Memory Limit: 64000K Total Submissions: 1667   Accepted: 532 Case Time Limit: 1000MS Description Frustrated at the number of distance queries required to find a reasonable route for his cow marathon, FJ decide

【点分治】poj1741 Tree / poj2114 Boatherds / poj1987 Distance Statistics

三道题都很类似.给出1741的代码 #include<cstdio> #include<algorithm> #include<cstring> using namespace std; #define MAXN 10001 typedef pair<int,int> Point; int n,K,ans; int v[MAXN<<1],w[MAXN<<1],first[MAXN],next[MAXN<<1],en; vo

POJ - 1987 Distance Statistics 树上的分治

题目大意:和poj 1741的那题和类似,求树上节点之间的距离小于等于k的节点对有多少对 解题思路:具体可参考:<分治算法在树的路径问题中的应用--漆子超> 给这题的输入坑了,注意输入,不然会超时 #include<cstdio> #include<vector> #include<algorithm> using namespace std; #define maxn 40010 int vis[maxn], Sum[maxn], d[maxn], dp[

poj1987 Distance Statistics

普通dfs访问每个点对的复杂度是O(n^2)的,显然会超时. 考虑访问到当前子树的根节点时,统计所有经过根的点(u, v)满足: dist(u) + dist(v) <= maxd,并且 belong(u)≠belong(v)(即u,v不在同一子树). 这里说的距离指的是节点到跟的距离. 可以用作差法,即用所有满足条件的点对数减去那些在根节点为当前子树根节点的儿子节点的点对数. 上面一步可以用O(nlogn)的复杂度解决,即先排序再比较. 根节点子树可以递归解决,用树的点分治. 总复杂度上界是O

POJ 1987 Distance Statistics (树上点分治)

题目地址:POJ 1987 点分治模板题,跟POJ 1741几乎一样,.. 代码如下: #include <iostream> #include <string.h> #include <math.h> #include <queue> #include <algorithm> #include <stdlib.h> #include <map> #include <set> #include <stdi

POJ 1987 Distance Statistics 树分治

Distance Statistics Description Frustrated at the number of distance queries required to find a reasonable route for his cow marathon, FJ decides to ask queries from which he can learn more information. Specifically, he supplies an integer K (1 <= K

树分治基础模板以及树的重心(poj1741 tree)

好久没有更新博文了,这里更新一发~~ Give a tree with n vertices,each edge has a length(positive integer less than 1001). Define dist(u,v)=The min distance between node u and v. Give an integer k,for every pair (u,v) of vertices is called valid if and only if dist(u,v

codechef Prime Distance On Tree(树分治+FFT)

题目链接:http://www.codechef.com/problems/PRIMEDST/ 题意:给出一棵树,边长度都是1.每次任意取出两个点(u,v),他们之间的长度为素数的概率为多大? 树分治,对于每个根出发记录边的长度出现几次,然后每次求卷积,用素数表查一下即可添加答案. 1 #include<algorithm> 2 #include<cstdio> 3 #include<cmath> 4 #include<cstring> 5 #include

【POJ1741】Tree 树分治 模板咯?

广告: #include <stdio.h> int main() { puts("转载请注明出处[vmurder]谢谢"); puts("网址:blog.csdn.net/vmurder/article/details/44302921"); } 题意: 给你一棵无根树,求有多少点对之间距离<=K. 题解: 树分治. 然后对于一个重心X,我们把它的所有子树中的所有点存到结构体数组中. 结构体中存距离和子树编号. 第一遍sort,我们双指针扫哪些点

POJ1741 经典树分治

题意:有一棵树,每条边有一个距离,求dis(u,v)<=k的点的对数 题解:树分治,对于一颗树上的两点,要么在同一颗子树上,要么在不同子树上,要么一个点是根,另一个在某一子树上,对于第一种情况我们可以通过递归来变成第二种或者第三种情况.我们对于某一颗子树来说我们先统计dis[u]+dis[v]<=k的点的对数,然后把该子树的所有子节点为根的这颗子树中dis[u]+dis[v]<=k的点的对数删去,因为在递归到后面会计算重复(这是第一种情况),所以这样就得到了该子树的满足条件的点对数,那么