洛谷 P1352 沒有上司的舞會
阿新 • • 發佈:2017-05-05
整數 urn read getc -s blog 計算 情況 def
題目描述
某大學有N個職員,編號為1~N。他們之間有從屬關系,也就是說他們的關系就像一棵以校長為根的樹,父結點就是子結點的直接上司。現在有個周年慶宴會,宴會每邀請來一個職員都會增加一定的快樂指數Ri,但是呢,如果某個職員的上司來參加舞會了,那麽這個職員就無論如何也不肯來參加舞會了。所以,請你編程計算,邀請哪些職員可以使快樂指數最大,求最大的快樂指數。
輸入輸出格式
輸入格式:
第一行一個整數N。(1<=N<=6000)
接下來N行,第i+1行表示i號職員的快樂指數Ri。(-128<=Ri<=127)
接下來N-1行,每行輸入一對整數L,K。表示K是L的直接上司。
最後一行輸入0 0
輸出格式:
輸出最大的快樂指數。
輸入輸出樣例
輸入樣例#1:7 1 1 1 1 1 1 1 1 3 2 3 6 4 7 4 4 5 3 5 0 0輸出樣例#1:
5
求出根參加/不參加宴會情況下子樹的快樂指數和,之後分類討論.
屠龍寶刀點擊就送
#include <ctype.h> #include <cstring> #include <cstdio> #define N 6500 void read (int&x) { x=0;bool f=0; char ch=getchar(); while (!isdigit(ch)) {if(ch==‘-‘) f=1;ch=getchar();} while (isdigit(ch)) x=x*10+ch-‘0‘,ch=getchar(); x=f?(~x)+1:x; } struct node { int next,to; }edge[N<<1]; int g[N],f[N],who,boss[N],R[N],n,head[N],cnt; void add(int u,intv) { edge[++cnt].next=head[u]; edge[cnt].to=v; head[u]=cnt; } int max(int a,int b) {return a>b?a:b;} void dfs(int x) { g[x]=R[x]; for(int i=head[x];i;i=edge[i].next) { dfs(edge[i].to); f[x]+=max(f[edge[i].to],g[edge[i].to]); g[x]+=f[edge[i].to]; } } int main() { read(n); for(int i=1;i<=n;i++) read(R[i]); for(int x,y;;) { scanf("%d%d",&x,&y); if(x==0&&y==0) break; add(y,x); boss[x]=y; } for(int i=1;i<=n;i++) if(!boss[i]) {who=i;break;} dfs(who); printf("%d",max(f[who],g[who])); return 0; }
洛谷 P1352 沒有上司的舞會