JSOI BZOJ4472salesman——树形dp

Description

某售货员小T要到若干城镇去推销商品,由于该地区是交通不便的山区,任意两个城镇
之间都只有唯一的可能经过其它城镇的路线。 小T 可以准确地估计出在每个城镇停留的净收
益。这些净收益可能是负数,即推销商品的利润抵不上花费。由于交通不便,小T经过每个
城镇都需要停留,在每个城镇的停留次数与在该地的净收益无关,因为很多费用不是计次收
取的,而每个城镇对小T的商品需求也是相对固定的,停留一次后就饱和了。每个城镇为了
强化治安,对外地人的最多停留次数有严格的规定。请你帮小T 设计一个收益最大的巡回方
案,即从家乡出发,在经过的每个城镇停留,最后回到家乡的旅行方案。你的程序只需输出
最大收益,以及最优方案是否唯一。方案并不包括路线的细节,方案相同的标准是选择经过
并停留的城镇是否相同。因为取消巡回也是一种方案,因此最大收益不会是负数。小T 在家
乡净收益是零,因为在家乡是本地人,家乡对小 T当然没有停留次数的限制。

Input

输入的第一行是一个正整数n(5<=n<=100000),表示城镇数目。城镇以1到n的数命名。小T 的家乡命
名为1。第二行和第三行都包含以空格隔开的n-1个整数,第二行的第i个数表示在城镇
i+1停留的净收益。第三行的第i个数表示城镇i+1规定的最大停留次数。所有的最大
停留次数都不小于2。接下来的n-1行每行两个1到n的正整数x,y,之间以一个空格
隔开,表示x,y之间有一条不经过其它城镇的双向道路。输入数据保证所有城镇是连通的。

Output

输出有两行,第一行包含一个自然数,表示巡回旅行的最大收益。如果该方案唯一,在
第二行输出“solution is unique”,否则在第二行输出“solution is not unique”。

Sample Input

9

-3 -4 2 4 -2 3 4 6

4 4 2 2 2 2 2 2

1 2

1 3

1 4

2 5

2 6

3 7

4 8

4 9
Sample Output

9

solution is unique

//最佳路线包括城镇 1,2, 4, 5, 9。


先BB两句,这题我被C++的优先队列坑了,我手调当q.size()=0时,q.top()居然还能返回数值,我也是醉了。
言归正传,这题要求我们求最大值,我们又发现要求的是最大值,所以果断树形dp。我们开一个优先队列,再dp时将它每一个儿子的值都扔到里面。因为某地最多能停留w[i]次,这就意味着可以取w[i]-1个子节点,如果节点<0,肯定是不取更优,如果==0,则可以取也可以不取,所以这样就输solution is not unique。如果w[i]==1并且剩下有两个权值一样的子节点,随便取哪个都行,所以也输solution is not unique。这样这道题就做完了。

Code:

#include<bits/stdc++.h>
#include<queue>
#define MAXN 100010
#define pa pair<int,int>
#define mp make_pair
#define fi first
#define se second
using namespace std;
int read(){
    char c;int x=0,y=1;while(c=getchar(),(c<'0'||c>'9')&&c!='-');
    if(c=='-') y=-1;else x=c-'0';while(c=getchar(),c>='0'&&c<='9')
    x=x*10+c-'0';return x*y;
}
int n,v[MAXN],w[MAXN],head[MAXN<<1],nxt[MAXN<<1],to[MAXN<<1],cnt,fa[MAXN];
void add(int x,int y){
    to[cnt]=y;nxt[cnt]=head[x];head[x]=cnt;cnt++;
    to[cnt]=x;nxt[cnt]=head[y];head[y]=cnt;cnt++;
}
void BuildTree(int x,int wy){
    fa[x]=wy;
    for(int i=head[x];i!=-1;i=nxt[i]){
        int go=to[i];
        if(go==fa[x]) continue;
        BuildTree(go,x);
    }
}
pa dfs(int x){
    priority_queue<pa> q;
    for(int i=head[x];i!=-1;i=nxt[i]){
        int go=to[i];
        if(go==fa[x]) continue;
        q.push(dfs(go));
    }
    int res=0,p=0,re=0;
    while(w[x]&&!q.empty()){
        pa a=q.top();q.pop();
        if(a.fi==0){p=1;break;}if(a.fi<0)break;
        if(a.se) p=1;re=a.fi;res+=a.fi;w[x]--;
        if(q.empty()) break;
        pa b=q.top();
        if(!w[x]&&b.fi==re) p=1;
    }
    while(!q.empty()) q.pop();
    return mp(res+v[x],p);
}
int main()
{
    memset(head,-1,sizeof(head));
    n=read();v[1]=0;w[1]=2e9;
    for(int i=2;i<=n;i++) v[i]=read();
    for(int i=2;i<=n;i++) w[i]=read(),w[i]--;
    for(int i=1;i<=n-1;i++){
        int x=read(),y=read();
        add(x,y);
    }
    BuildTree(1,0);
    pa ans=dfs(1);
    printf("%d\n",ans.fi);
    if(ans.se) puts("solution is not unique");
    else puts("solution is unique");
    return 0;
}

猜你喜欢

转载自blog.csdn.net/stevensonson/article/details/79834343