Luogu P2014 选课
第一次写博客,紧张。
题目描述
在大学里每个学生,为了达到一定的学分,必须从很多课程里选择一些课程来学习,在课程里有些课程必须在某些课程之前学习,如高等数学总是在其它课程之前学习。现在有N门功课,每门课有个学分,每门课有一门或没有直接先修课(若课程a是课程b的先修课即只有学完了课程a,才能学习课程b)。一个学生要从这些课程里选择M门课程学习,问他能获得的最大学分是多少?
输入输出格式
输入格式:
第一行有两个整数N,M用空格隔开。(1<=N<=300,1<=M<=300)
接下来的N行,第I+1行包含两个整数ki和si, ki表示第I门课的直接先修课,si表示第I门课的学分。若ki=0表示没有直接先修课(1<=ki<=N, 1<=si<=20)。
输出格式:
只有一行,选M门课程的最大得分。
输入样例#1:
7 4
2 2
0 1
0 4
2 1
7 1
7 6
2 2
输出样例#1:
13
树形dp:在树的数据结构上做动态规划,通过有限次的遍历树,以求解问题。其实我动归学的一点都不好
基础题目见MZOJ1264longest、MZOJ1063士兵守卫。
- 将各课程与其先修课程用邻接表连接(双向、单向边都可以?)。所有无先修课程的科目都连上0,即0作为根节点。
- now代表当前节点;fa代表其父亲;size代表截至当前子树,总共有多少节点,并用j(<size)算出取各种数量的科目的学分;f[x][y]代表截至x结点,总共选取y个结点可得到的最多学分。
- 从(0,-1)开始dp,一个子树一个子树地求出各结点的f值。我们将整个树分为两部分,一部分为当前的子树,从中选取k个点;另一部分为其他所有遍历过的子树,从中选取j-k个点。这样依次求出各种情况的f,即f[now][j]=max(f[now][j],f[to][k]+f[now][j-k])。
- 每个点本身可以选择不学习,其子节点当然也不能学习,则不能得到相应的学分,即f[now][0]=0;若选择学习此结点,则可获得本课程的学分(w[now]),即f[now][1]=w[now]。再用相同子问题递推求出其子节点的f值。
代码:
#include<bits/stdc++.h>
using namespace std;
#define read read()
const int maxn=305;
int n,m,cnt;
int head[maxn],w[maxn],f[maxn][maxn];
inline int read
{
int x=0,f=1;char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-') f=-1;ch=getchar();}
while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
return x=x*f;
}
struct edge
{
int v,nxt;
}e[maxn<<1];
void add(int u,int v)
{
e[cnt].v=v;
e[cnt].nxt=head[u];
head[u]=cnt++;
}
void readdata()
{
int j;
memset(head,-1,sizeof(head));
n=read;m=read;m++;
for(int i=1;i<=n;i++)
{
j=read;w[i]=read;
add(i,j);
add(j,i);
}
}
int dp(int now,int fa)
{
int s=0,size=1;
f[now][0]=0;f[now][1]=w[now];
for(int i=head[now];~i;i=e[i].nxt)
{
int to=e[i].v;
if(to==fa) continue;
s=dp(to,now);
size+=s;
for(int j=size;j>0;j--)
{
for(int k=0;k<j;k++)
{
if(k>s) break;
f[now][j]=max(f[now][j],f[to][k]+f[now][j-k]);
}
}
}
return size;
}
int main()
{
//freopen("input.txt","r",stdin);
readdata();
dp(0,-1);
printf("%d",f[0][m]);
return 0;
}