树形DP,f[j][i]表示有j个robot遍历i及其子树并且在此子树返回地球所需的最小值,但还有k个robot遍历子树i及其子树,再返回父亲节点m个机器人这种情况,但是可以证明这种情况一定不会是最优解,所以不予考虑。还有一个地方就是f[0][i]表示用0个robot遍历意思就是说用n个机器人遍历此子树,再返回,可知n=1,道理和前面那种情况的道理一样。在求f[i][j]是用的是分组背包的思想,刚开始多用了一个数组g[][],来实现这个过程,后来看了其他人的代码,想了一想,可以用滚动数组优化,用f[i][j]就可以实现这个过程了
#include <iostream> #include <cstdio> #include <cstring> #define LL long long using namespace std; const int maxn=10000+10; const LL inf=0x3f3f3f3f3f3f3f3f; int e[2*maxn],head[maxn],cost[maxn*2],next[maxn*2],vis[maxn]; int n,s,k,tot; LL f[maxn][11],g[maxn][11]; void dp(int x) { int i=head[x]; vis[x]=1; if(next[i]==-1&&x!=s) return;//注意x!=s这一条件因为根可能只有一个子树 int p,q,w,tv; int j; for(;i!=-1;i=next[i]) { tv=e[i]; if(!vis[tv]) { dp(tv); for(j=k;j>=0;j--) { f[x][j]=f[x][j]+f[tv][0]+2*cost[i]; for(q=1;q<=j;q++) f[x][j]=min(f[x][j],f[x][j-q]+f[tv][q]+q*cost[i]); } } } } int main() { while(~scanf("%d%d%d",&n,&s,&k)) { tot=0; memset(head,-1,sizeof(head)); memset(vis,0,sizeof(vis)); memset(f,0,sizeof(f)); int i,u,v,c; for(i=0;i<n-1;i++) { scanf("%d%d%d",&u,&v,&c); e[tot]=v; cost[tot]=c; next[tot]=head[u]; head[u]=tot++; e[tot]=u; cost[tot]=c; next[tot]=head[v]; head[v]=tot++; } dp(s); printf("%I64d\n",f[s][k]); } return 0; }