【SSL】1565将功补过

【SSL】1565将功补过

Time Limit:10000MS Memory Limit:65536K
Case Time Limit:1000MS

Description

作为间谍专家的Elvis Han受窃取X星球军事中心的秘密情报，他已经成功进入军事中心。但是很不幸的是，在他还没有找到任务需要情报的时候就被发现，这时他清楚他不可能完成任务了，不过还有机会将功补过，也就是得到一些不如任务情报有价值的其他情报，如果得到的情报的总价值大于等于任务情报价值，他也不会受到惩罚。很幸运的是他已经得到的军事中心的地图，情报都是隐藏在各个道路上的，但是他只有时间遍历一定数量的路（时间宝贵呀！还要逃跑。。）现在你做为他的助手，给你地图和每个道路情报价值，希望你分析出，看他能不能将功补过。
　　军事中心是一个严格的二叉树，也就是说，如果有个点可以分道，一定是分出，也只分出2条道路，现在Elvis Han正处在第一个分道处，也就是说树的根结点处。每条道路上都有一个分数，就是这个道路上的情报价值。但是他只有时间走M条路，他的最终情报价值总和就是他所经过的路的情报价值总和（假设他到过的路一定可以把所有情报得到）希望你给出一个方案使得他可以尽量多地获取情报以便将功补过。

Input

共有N行：
第一行：3个数据：N，M，Q（N表示有多少个路口，包括分道和不分道的路口；M表示他可以有时间走的道路总数；Q表示他的任务情报的价值）
第2~N行：每行3个数据，Xi,Yi,Wi (X，Y表示第I条道路连接的2个路口，W表示这条道路上的情报价值分, 注意，所有数据均在Lonint范围内)

Output

共包含2行：
第一行：输出TRUE/FALSE（注意大小写），表示他是否可以收集够任务情报价值
第二行：输出一个数据：
如果他可以完成任务，就输出他收集的情报总价值超过任务情报价值的部分。（正数）
如果不能完成任务，就输出一个数，表示他不能还差多少分才够任务情报价值。（负数）

Sample Input

【样例输入1】

3 1 10
1 2 10
1 3 8

【样例输入2】

9 3 49
6 2 15
7 2 10
8 7 6
7 9 15
1 3 20
2 1 10
4 3 8 
3 5 7

Sample Output

【样例输出1】

TRUE
0

【样例输出2】

FALSE
-4

Hint

<数据规模>
对于30%的数据 保证有N<=10
对于50%的数据 保证有N<=40
对于全部的数据 保证有 N<=100

思路

树状DP
f[i][j]表示第i个节点选j条边的价值
f[i][j]=max(f[i][j-k-1]+f[i的子节点][k]+该边权值)
f[i][0]=0
0<=k<j<=m

代码

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
long long n,m,q,w[1010],f[1010][1010],head[1010],e[1010],nxt[1010];
bool vis[1010];
void input()
{
	long long i,x,y,v,tot=0;
	memset(f,-1,sizeof(f));
    for(scanf("%lld%lld%lld",&n,&m,&q),i=1;i<n;i++)
	{
		scanf("%lld%lld%lld",&x,&y,&v);
		e[++tot]=y,nxt[tot]=head[x],head[x]=tot,w[tot]=v;
		e[++tot]=x,nxt[tot]=head[y],head[y]=tot,w[tot]=v;
	}
    return;
}
void TREEDP(long long x)//树状DP 
{
	long long i,j,k;
	
	for(f[x][0]=0,vis[x]=1,i=head[x];i;i=nxt[i])
		if(!vis[e[i]])
			for(TREEDP(e[i]),j=m;j>=0;j--)
				for(k=0;k<j;k++)
					f[x][j]=max(f[x][j],f[x][j-k-1]+f[e[i]][k]+w[i]);
	return;
}
int main()
{
	input(),TREEDP(1);
	if(f[1][m]>=q)printf("TRUE\n%lld",f[1][m]-q);
	else printf("FALSE\n%lld",f[1][m]-q);
	return 0;
}