P1352 沒有上司的舞會
阿新 • • 發佈:2017-06-30
編號 -m += 表示 輸出 參加 格式 cstring 之間
題目描述
某大學有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
我們用dp[i][0]表示這個點不參加
用dp[i][1]表示這個點參加
1 #include<iostream> 2 #include<cstdio> 3 #include<cstring> 4 #include<cmath> 5 #include<queue> 6 using namespacestd; 7 const int MAXN=6001; 8 void read(int &n) 9 { 10 char c=‘+‘;int x=0;bool flag=0; 11 while(c<‘0‘||c>‘9‘) 12 {c=getchar();if(c==‘-‘)flag=1;} 13 while(c>=‘0‘&&c<=‘9‘) 14 {x=x*10+(c-48);c=getchar();} 15 flag==1?n=-x:n=x; 16 } 17 int n; 18 intha[MAXN]; 19 int fa[MAXN]; 20 int ch[MAXN]; 21 int dp[MAXN][3]; 22 int deg[MAXN]; 23 int ans=0; 24 void Topsort() 25 { 26 queue<int>q; 27 for(int i=1;i<=n;i++) 28 if(deg[i]==0) 29 q.push(i); 30 while(q.size()!=0) 31 { 32 int p=q.front(); 33 q.pop(); 34 deg[fa[p]]--; 35 if(deg[fa[p]]==0&&fa[p]!=-1) 36 q.push(fa[p]); 37 dp[fa[p]][0]+=max(dp[p][0],dp[p][1]); 38 // 上司不參加 39 dp[fa[p]][1]+=dp[p][0]; 40 // 參加 41 if(fa[p]==-1) 42 ans+=max(dp[fa[p]][1],dp[fa[p]][0]); 43 } 44 printf("%d",ans); 45 46 } 47 int main() 48 { 49 read(n); 50 memset(ch,-1,sizeof(ch)); 51 memset(fa,-1,sizeof(fa)); 52 for(int i=1;i<=n;i++) 53 read(ha[i]); 54 for(int i=1;i<=n;i++)dp[i][1]=ha[i]; 55 for(int i=1;i<=n;i++) 56 { 57 int x,y; 58 read(x);read(y); 59 fa[x]=y; 60 ch[y]=x; 61 deg[y]++; 62 } 63 Topsort(); 64 return 0; 65 }
P1352 沒有上司的舞會