2020.6.28
终于找到了一个合适的学习场所,然而起太早貌似今天上午效率并不高,一脸吃了三道题wa,然后后来一道题思路是对的,但是看了题解才过,果然两天不做题就手生。
下学期基本就是这样了,预计我的一天就是,早上在家锻炼,然后一天泡在图书馆,晚上回来该玩就玩,也挺好,不指望学校能把我们安排在国科大了。
这道题的题意是问,大学很多课都有先修课,上完才能上这节课,比如计网 ,然后问最多选k门课的最大得分,显然k门课要按照一定顺序选才合法,而且显然并非所有的链上的子树都能满足最优解。给出的边仅为n条,所以显而易见这道题是树型dp(废话)。那么状态转移方程,我一开始想的是用背包,即f[j] = max(f[j], f[j - a[I]] + val[i]),但后来想了想缺了一个约束条件,显然单就选的数量来看是不存在直接的从上一维直接转移过来的,我们需要明确经过了哪些点,哪些呢?当然是子树的状态啦,于是就有
dp[u][p] = max(dp[u][p] , dp[u][p - t] + dp[v][t]);
本题还有个问题就是需要处理下源点,将所有入度为0的点连到0上,然后这样问题就转化为在0为根的树选取k + 1个课了。我看题解的时候其实已经推到上面这一步了,后来发现是我的推的顺序出锅了hhhh
代码:
#include <bits/stdc++.h>
using namespace std;
#define limit (1000 + 5)//防止溢出
#define INF 0x3f3f3f3f
#define inf 0x3f3f3f3f3f
#define lowbit(i) i&(-i)//一步两步
#define EPS 1e-6
#define FASTIO ios::sync_with_stdio(false);cin.tie(0);
#define ff(a) printf("%d\n",a );
#define pi(a,b) pair<a,b>
#define rep(i, a, b) for(int i = a; i <= b ; ++i)
#define per(i, a, b) for(int i = b ; i >= a ; --i)
#define mint(a,b,c) min(min(a,b), c)
#define MOD 988244353
#define FOPEN freopen("C:\\Users\\tiany\\CLionProjects\\acm_01\\data.txt", "rt", stdin)
#define FOUT freopen("C:\\Users\\tiany\\CLionProjects\\acm_01\\dabiao.txt", "wt", stdout)
typedef long long ll;
typedef unsigned long long ull;
ll read(){
ll sign = 1, x = 0;char s = getchar();
while(s > '9' || s < '0' ){if(s == '-')sign = -1;s = getchar();}
while(s >= '0' && s <= '9'){x = x * 10 + s - '0';s = getchar();}
return x * sign;
}//快读
void write(ll x){
if(x < 0) putchar('-'),x = -x;
if(x / 10) write(x / 10);
putchar(x % 10 + '0');
}
int n,k;
int head[limit],cnt;
void init(){
memset(head, -1, sizeof(head));
cnt = 0;
}
struct node{
int to, next;
}edge[limit<<1];
void add(int u, int v){
edge[cnt].to = v;
edge[cnt].next = head[u];
head[u] = cnt++;
}
int val[limit];
int dp[limit][limit];//第i个节点位置上选j个的
void dfs(int u,int pre){
dp[u][1] = val[u];
for(int i = head[u] ; ~i ; i = edge[i].next){
int v = edge[i].to;
if(v != pre){
dfs(v,u);
per(p ,1, k){
per(t, 1, p - 1){
dp[u][p] = max(dp[u][p], dp[u][p - t] + dp[v][t]);
}
}
}
}
}
int main() {
#ifdef LOCAL
FOPEN;
#endif
n = read(), k = read();
init();//初始化
rep(i ,1,n){
int x = read(), y = read();
val[i] = y;
add(x,i);
add(i,x);
}//添加边权
++k;
dfs(0 , -1);
write(dp[0][k]);
return 0;
}