码迷,mamicode.com
首页 > 其他好文 > 详细

POJ-3585-Accumulation-Degree

时间:2019-04-20 21:09:51      阅读:106      评论:0      收藏:0      [点我收藏+]

标签:并且   [1]   namespace   struct   详细   ons   ble   它的   回溯   

题目地址

题目大意:

一棵树。

· 树的每个边都具有正边权,代表边的容量。
· 树中度为1的点被命名为出海口。
· 每个边的流量不能超过容量。

A(x)是将点x视作一个无线喷水机,表示点x可以流到其他(如果他也是出海口,则排除他)出海口的最大流量。
你的任务找一个点,使这个最佳最大流量,输出这个值。

示例:

技术图片
A(1)= 11 + 5 + 8 = 24
详细说明:
1->2 = 11
1->4->3 = 5
1->4->5 = 8(因为1-> 4的容量为13)

A(2)= 5 + 6 = 11
详细说明:
2->1->4->3 = 5
2->1->4->5 = 6

A(3)= 5
细节:
3->4->5 = 5

A(4)= 11 + 5 + 10 = 26
细节:
4->1->2 = 11
4->3 = 5
4->5 = 10

A(5)= 10
细节:
5->4->1->2 = 10

最大流量就是A(4)=26。

输入

输入的第一行是整数T,表示测试数据的数量。
每个测试数据的第一行是正整数n。以下n-1行中的每一行包含由空格分隔的三个整数x,y,z,表示在节点x和节点y之间存在边缘,并且边缘的容量为z。节点编号从1到n。
所有元素都是非负整数,不超过200000.您可以假设测试数据都是树度量。

输出

对于每个测试用例,将结果输出到一行。

思路

一般来说,做这种比较没有头绪的题一般可以先写一遍暴力,然后再寻找一下规律。

暴力

思路很容易想到:
枚举每一个点,找出它的最大流量。

那么这个最大流量怎么求呢?
技术图片
比如找A(1),那么他流13的量到4,但是到出水口的水只有10的量,所以1->4这边最大量为10。
那么,我们可以用dfs回溯来完成这个过程,取叶子结点能流的最大量父节点的最小值即可。
一旦我们搜索到出海口,直接返回它邻边的值即可。
特别注意,边界条件要放在最后。不然如果找出海口的流量就会直接返回,向下列示例就会暴毙
技术图片
跑A(1)的时候应该是2,但是如果把边界条件写前面就会变成30。

int dfs(int x,int l){//x为点、l为上一个点
    int p=0;//p表示叶子结点能流向海里的最大量
    for(int j=head[x];j;j=e[j].next){//它的连边
        int v=e[j].to;
        if(v==l)continue;//防止死循环
        p+=min(e[j].dis,dfs(v,x));//这个边的量
    }
    if(du[x]!=1)return d[x]=p;//du为每个点的度
    else return e[head[x]].dis;
}

但是显而易见,数据范围给的是(N<=200000),如果这么做的时间复杂度为O(N^2),一定会TLE。
所以,我们必须把时间复杂度控制在可控范围内。

想法

这道题的dp看似无迹可寻,但似乎除了dp方法,没有别的可以完成这个复杂度了。
于是,我们思考一下,能否通过跑一遍dfs来推出其他A(x)的值?
我们注意到每次递归时有p的值,存在d数组里,思考一下是否有用?

试着跑从节点1跑一遍dfs,沿途把p用d数组保存起来。
d[i]代表i点下叶子结点的最大流量,当然出海口的就为0。

int dfs_(int x,int l){
    int p=0;d[x]=0;
    for(int j=head[x];j;j=e[j].next){
        int v=e[j].to;
        if(v==l)continue;
        p+=min(e[j].dis,dfs_(v,x));
    }
    if(du[x]!=1)return d[x]=p; 
    else return e[head[x]].dis;
}

用示例跑一遍,把d[i]输出:
技术图片

//dp[i]代表A(i)
dp[1]=24
dp[2]=11
dp[3]=5
dp[4]=26
dp[5]=10
//d[i]代表i点下叶子结点的最大流量
d[1]=24
d[2]=0
d[3]=0
d[4]=15
d[5]=0

我们尝试着用dp[1]推出他相邻的点dp[2],用到dp[1]、d[2]。
dp[2]由它的叶子结点的流量、父节点连出海口的量组成。
叶子结点,显而易见是:

d[2]

那么父节点后面的水量呢?
首先,**因为点1到点2只能流过不超过边权(1,2)的量。
所以:点1的叶子结点的最大量为:

min(边权(1,2),d[2])。
那么点1的父节点的量就是:dp[1]-min(边权(1,2),d[2]。

如果延伸到2,因为他们俩相连,2要先到1,才能通过1到1后面的出海口,所以水量不能超过他们俩的边权。

得推导公式:

min(边权(1,2),dp[1]-min(边权(1,2),d[2])

归纳:
任意相连的点(u,v)、已知dp[u]、d[v]、边权(u,v),求出dp[v]。

dp[v]=d[v]+min(边权(u,v),dp[u]-min(边权(u,v),d[v]));

但是,还是有漏洞的,如果一开始跑dfs的是出海口,那么这个式子就会炸掉,dp[u]为0,他再减一个数字就变成负数了...这种问题很好解决,我们特判一下就行啦:

if(du[x]==1)dp[v]=e[j].dis+d[v];
else dp[v]=d[v]+min(e[j].dis,dp[x]-min(e[j].dis,d[v]));

最终,我们的推导dfs也很容易写出来啦:

void dfs(int x,int l){////x为点、l为上一个点
    for(int j=head[x];j;j=e[j].next){//找相邻点
        int v=e[j].to;
        if(v==l)continue;//防死循环
        if(du[x]==1)dp[v]=e[j].dis+d[v];
        else dp[v]=d[v]+min(e[j].dis,dp[x]-min(e[j].dis,d[v]));
        ans=max(ans,dp[v]);//找出最大的A[i]
        dfs(v,x);//继续推导
    }
}

总结

所以,我们只需要跑两遍dfs,第一遍算出每个点的叶子结点连得最大量d[i]、A[1]。再通过第二遍dfs逐步推出每一个A[i]即可。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <vector>
#include <cmath>
using namespace std;
const int N=200001;
int t,n,u,v,w,head[N],numE=0,d[N],du[N],dp[N],ans,s;
struct Edge{
    int next,to,dis;
}e[2*N];
void addEdge(int from,int to,int dis){
    e[++numE].next=head[from];
    e[numE].to=to;
    e[numE].dis=dis;
    head[from]=numE;
}
int dfs_(int x,int l){
    int p=0;d[x]=0;
    for(int j=head[x];j;j=e[j].next){
        int v=e[j].to;
        if(v==l)continue;
        p+=min(e[j].dis,dfs_(v,x));
    }
    if(du[x]!=1)return d[x]=p; 
    else return e[head[x]].dis;
}
void dfs(int x,int l){
    for(int j=head[x];j;j=e[j].next){
        int v=e[j].to;
        if(v==l)continue;
        if(du[x]==1)dp[v]=e[j].dis+d[v];
        else dp[v]=d[v]+min(e[j].dis,dp[x]-min(e[j].dis,d[v]));
        ans=max(ans,dp[v]);
        dfs(v,x);
    }
}
int main() {
    scanf("%d",&t);
    while(t--){
        memset(head,0,sizeof(head));
        memset(d,0,sizeof(d));
        memset(dp,0,sizeof(dp));
        memset(du,0,sizeof(du));
        numE=0;ans=0;
        scanf("%d",&n);
        for(int i=1;i<n;i++){
            scanf("%d%d%d",&u,&v,&w);
            addEdge(u,v,w);addEdge(v,u,w);
            du[u]++;du[v]++;
        }
        dp[2]=dfs_(1,-1);
        dfs(1,-1);
        printf("%d\n",max(dp[1],ans));
    }
    return 0;
}

POJ-3585-Accumulation-Degree

标签:并且   [1]   namespace   struct   详细   ons   ble   它的   回溯   

原文地址:https://www.cnblogs.com/dmoransky/p/10742624.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!