CJOJ 2171 火車站開飯店（樹型動態規劃）

Description

政府邀請了你在火車站開飯店，但不允許同時在兩個相連的火車站開。任意兩個火車站有且只有一條路徑，每個火車站最多有 50 個和它相連接的火車站。
告訴你每個火車站的利潤，問你可以獲得的最大利潤為多少？
例如下圖是火車站網絡：

最佳投資方案是 1 ， 2 ， 5 ， 6 這 4 個火車站開飯店可以獲得的利潤為 90.

Input

第一行輸入整數 N(<=100000), 表示有 N 個火車站，分別用 1,2,……..,N 來編號。
接下來 N 行，每行一個整數表示每個站點的利潤，接下來 N-1 行描述火車站網絡，每行兩個整數，表示相連接的兩個站點。

Output

輸出一個整數表示可以獲得的最大利潤。

Sample Input

6
10
20
25
40
30
30
4 5
4 6
3 4
1 3
2 3

Sample Output

90

Http

CJOJ:http://oj.changjun.com.cn/problem/detail/pid/2171

Source

樹型動態規劃

解決思路

這道題與POJ2342真是有異曲同工之妙，這里不再過多敘述，請參考這里

代碼

#include<iostream> #include<cstdio> #include<cstdlib> #include<cstring> #include<algorithm> #include<vector> using namespace std;const int maxN=100001; const int inf=2147483647;int n; int Value[maxN]; vector<int> E[maxN]; int F[maxN][5]={0}; bool vis[maxN]={0};void dfs(int u);int main() {int u,v;cin>>n;for (int i=1;i<=n;i++)scanf("%d",&Value[i]);for (int i=1;i<n;i++){scanf("%d%d",&u,&v);E[u].push_back(v);E[v].push_back(u);}dfs(1);cout<<max(F[1][1],F[1][0])<<endl;return 0; }void dfs(int u) {vis[u]=1;F[u][1]=Value[u];F[u][0]=0;for (int i=0;i<E[u].size();i++){int v=E[u][i];if (vis[v]==0){dfs(v);F[u][1]+=F[v][0];F[u][0]+=max(F[v][1],F[v][0]);}}return; }

轉載于:https://www.cnblogs.com/SYCstudio/p/7138199.html

總結

以上是生活随笔為你收集整理的CJOJ 2171 火车站开饭店（树型动态规划）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。