• poj 1741 (树的分治)


    题意:给定一棵N(1<= N <=10000)个结点的带权树,定义dist(u,v)为u,v两点间的最短路径长度,路径的长度定义为路径上所有边的权和。再给定一个 K  ,如果对于不同的两个结点a,b,如果满足dist(a,b) <=K,则称(a,b)为合法点对。求合法点对个数。

    思路:看了论文《分治算法在树的路径问题中的应用》,里面讲解的很清楚,一条路径要么过根节点,要么在一颗子树中,所以用分治算法。找到树的重心作为根节点,这样每次树的节点数至少减少一半。处理经过当前根节点路径<=k的点对数,然后把根节点去掉后就把原来的树分成几颗子树了,再处理子树。我们在求经过一个根节点的路径时,里面还包含了点对属于同一颗子树的情况,所以要去掉这部分的点。

    dis(i)+dis(j)<=k(i,j的父节点不为根节点的同一个儿子)

    =dis(i)+dis(j)<=k-dis(i)+dis(j)<=k(i,j的父节点属于根节点的同一儿子).





    #include <algorithm>
    #include<stdio.h>
    #include<string.h>
    const int N=10010;
    using namespace std;
    int head[N],num,f[N],son[N],n,D,root,size,ans,dis[N],d[N],cum;
    bool vis[N];
    #define max(a,b) (a<b?b:a)
    struct edge
    {
    	int st,ed,w,next;
    }e[N*2];
    void addedge(int x,int y,int w)
    {
    	e[num].st=x;e[num].ed=y;e[num].w=w;e[num].next=head[x];head[x]=num++;
    	e[num].st=y;e[num].ed=x;e[num].w=w;e[num].next=head[y];head[y]=num++;
    }
    void getroot(int u,int father)//求树的重心
    {
        int i,v;
    	f[u]=0;son[u]=1;
    	for(i=head[u];i!=-1;i=e[i].next)
    	{
    		v=e[i].ed;
    		if(vis[v]||v==father)continue;
    		getroot(v,u);
    		son[u]+=son[v];
    		f[u]=max(f[u],son[v]);
    	}
    	f[u]=max(f[u],size-son[u]);
    	if(f[u]<f[root])root=u;
    }
    void getdis(int u,int father)//求节点到根节点的距离
    {
    	int i,v;
    	son[u]=1;//更新子树的节点的子节点数,不更新也能ac
    	d[cum++]=dis[u];//将点到根节点的距离加入数组
    	for(i=head[u];i!=-1;i=e[i].next)
    	{
    		v=e[i].ed;
    		if(vis[v]||v==father)continue;
    		dis[v]=dis[u]+e[i].w;
    		getdis(v,u);
    		son[u]+=son[v];
    	}
    }
    int cont(int u,int mit)
    {
    	int res=0,L,R;
    	dis[u]=mit;
    	cum=0;
    	getdis(u,0);
    	sort(d,d+cum);//将点到根节点的距离排序
    	for(L=0,R=cum-1;L<R;)
    	{
    		if(d[L]+d[R]<=D)//如果d[L]+d[R]<=D,L代表的节点可以与(R-L)个节点成对
    			res+=(R-L++);
    		else R--;
    	}
    	return res;
    }
    void work(int u)
    {
    	int i,v;
       vis[u]=true;
       ans+=cont(u,0);//路径经过该根节点的点对数
       for(i=head[u];i!=-1;i=e[i].next)
       {
          v=e[i].ed;
    	  if(vis[v])continue;
    	  ans-=cont(v,e[i].w);//减去属于v子树的点对数
    	  root=0;f[root]=size=son[v];
    	  getroot(v,0);//求v子树的根节点
    	  work(root);//求v子树的点对
       }
    }
    int main()
    {
    	int i,x,y,w;
    	while(scanf("%d%d",&n,&D),n||D)
    	{
    		memset(head,-1,sizeof(head));
    		num=0;
    		for(i=1;i<n;i++)
    		{
    			scanf("%d%d%d",&x,&y,&w);
    			addedge(x,y,w);
    		}
    		memset(vis,false,sizeof(vis));
    		root=0;f[root]=size=n;ans=0;
    		getroot(1,0);
    		work(root);
    		printf("%d
    ",ans);
    	}
    	return 0;
    }
    


  • 相关阅读:
    javascript删除JSON元素
    State ,Getter , Mutation , Action , Module
    mutation和action区别
    图片懒加载
    git命令合并分支代码
    vue 项目生产环境下去重console
    rem px pt em (区别)
    npm 安装依赖 以及dependencies 和 devDependencies的区别
    仿百度搜索
    h5页面在移动端需要注意的一些事情
  • 原文地址:https://www.cnblogs.com/keanuyaoo/p/3356122.html
Copyright © 2020-2023  润新知