POJ3345 Bribing FIPA 树形DP

Problem Address：http://poj.org/problem?id=3345

【前言】

前面的两篇都是把树转化为二叉树计算，虽然简单，但是不是很直观。

这份代码是直接用邻接表做。用vector记录邻接的边。

然后对每个结点直接进行背包。

相比起记忆化搜索，直接背包的好处就是少了很多次函数调用，应该是节省了不少时间。

【思路】

这道题比较难处理的是输入。

gets()读取整行数据。

sscanf用来从字符串读入。

stringstream可以用来设置输入流。

然后用map映射结点的索引号。

以0结点作为根节点，连接到每个原来的根。

先对子结点进行背包，再对父结点进行背包。

同时需要更新父结点。如果父结点在包含所有子结点时的费用小于当前费用，则需更新。

【代码】

#include <iostream>#include <map>#include <sstream>#include <vector>#include <string>using namespace std;const int maxn = 200;const int MAX = 99999999;map<string, int> ma;vector<int> adj[maxn+5];bool visited[maxn+5];int diamond[maxn+5];int dp[maxn+5][maxn+5];int n, m;int solve(int v){visited[v] = true;int i, j, k;int t, u;int num = 1;for (i=0; i<=n; i++)dp[v][i] = MAX;dp[v][0] = 0;for (i=0; i<adj[v].size(); i++){u = adj[v][i];if (visited[u]) continue;num += solve(u);for (j=n; j>0; j--){for (k=1; j-k>=0; k++){if (dp[v][j-k]+dp[u][k]<dp[v][j])dp[v][j] = dp[v][j-k] + dp[u][k];}}}if (dp[v][num]>diamond[v])dp[v][num] = diamond[v];return num;}int main(){char str[1000];char name[105];int index;int i, j, tn, tc;while(gets(str)){if (str[0]=='#') break;sscanf(str, "%d %d", &n, &m);index = 1;ma.clear();for (i=0; i<=n; i++){adj[i].clear();visited[i] = false;}for (i=0; i<n; i++){scanf("%s", name);if (ma.find(name)==ma.end()){ma[name] = index;index++;}tn = ma[name];scanf("%d", &diamond[tn]);gets(str);stringstream ss(str);while(ss>>name){if (ma.find(name)==ma.end()){ma[name] = index;index++;}tc = ma[name];adj[tn].push_back(tc);adj[tc].push_back(tn);if (!visited[tc])visited[tc] = true;}}diamond[0] = MAX;for (i=1; i<=n; i++){if (!visited[i])adj[0].push_back(i);}for (i=0; i<=n; i++)visited[i] = false;solve(0);int ans = MAX;for (i=m; i<=n; i++)if (dp[0][i]<ans)ans = dp[0][i];printf("%d\n", ans);}return 0;}