Luogu P2014 选课

题面

对于这道题,我们考虑在树形dp上套背包。我们会非常自然的采用dfs扫描整棵树,然后对树上的每个节点都进行一次背包。

计\(dp[i][j]\)为在以第\(i\)号节点为根结点的子树中,用题目中选法选取\(j\)项的最大值。

我们在dfs的过程中,采用递归的方式,在子节点都处理完之后,便考虑将所有子节点的答案综合,得到当前节点的答案。

很显然,就是在容量为\(j\)的01背包中放下\(i\)节点的所有子节点背包中的答案,我们很容易想到下面的DP方程
\[f[x][j]=max(f[to][k]+f[x][j-k])\]
\(x\)为当前节点,\(to\)为它的某个孩子节点,\(j\)和\(k\)是枚举的两个变量。

就得到了

for(int i=0;j<G[x].size();i++){
    int to=G[x][i];
    dfs(to,x);
    for(int j=m+1;j>=1;j--){
        for(int k=m;k>=0;k--){
            if(j-k<1) continue;
            f[x][j]=max(f[x][j],f[to][k]+f[x][j-k]);
        }
    }
}

最终的答案就是\(f[0][m+1]\)了,因为所有没有前提条件的课程都可以指向\(0\),即让第\(0\)号课程成为他们的先决条件,因为\(0\)号课程自身也算一个课程,所以第二项就是\(m+1\)了

优化

显然,上述算法时间复杂度为\(O(N*M^2)\),虽然在本题已经可以通过,但是如果数据量增加到\(n<=2000\)时,应该如何应对

由于对于每一个节点上的背包,我们每一次都枚举到了\(m+1\),但是大部分情况下实际背包一般不会那么大,所以在时间上会产生很大的开销。

考虑下面的这份代码

int dfs(int node)
{
    int sum=0;
    f[node][1]=cost[node];
    for(int i=0;j<G[x].size();i++){
        int to=G[node][i];
        int cnt=dfs(to);
        sum+=cnt;
        for(int j=m+1;j>=1;j--)
            for(int k=min(cnt,j);k>=0;k--){
                if(j<k+1) continue;
                f[node][j]=max(f[node][j],f[to][k]+f[node][j-k]);
            }
    }
    sum++;
    return sum;
}

用\(sum\)存放当前节点为根的子树的大小,同时dfs值为子树大小直接进行传值。

显然背包的第二重循环是可以像这样优化的,毕竟子节点产生的序列大小也才那么点,这样的话整个子树上的所有情况都只会在这个节点上体现一次(显然之前的方法大量冗余的计算会使得它制造的无用情况会大大多于现在),所以整棵树一共是\(n\)个节点,对于每个节点它的孩子的若干个\(m\)情况只会经过一次,并且在\(O(1)\)时间内直接取出每组情况最优值,所以对于每个点的复杂度是\(O(m)\)的,所以总的算法时间复杂度就为\(O(N*M)\)了,就可以对付增强后的数据了。

完整代码



转载自本人Luogu Blog这篇文章

原文地址:https://www.cnblogs.com/macesuted/p/solution-P2014.html

时间: 2024-11-07 16:28:23

Luogu P2014 选课的相关文章

MZOJ 1132 &amp;&amp; LuoGu P2014 选课

MZOJ 1132 && LuoGu P2014 选课    [传送门] #include<bits/stdc++.h> #define maxn 100000 #define maxm 500 using namespace std; int k=0,head[maxn]; int f[maxm][maxm],w[maxn]; int ans=0; int N,M; struct node{ int v,nxt; }e[maxn<<1]; void adde(in

Luogu P2014 选课 题解报告

题目传送门 [题目大意] 有n门选修课,每一门课都有固定的学分$S_i$,每个学生可以选m门课.有些选修课有先修课,每一门课最多只有一门先修课,求能获得的最多学分. [思路解析] 设f[x][t]表示在以x结点为根的子树中选t门课能获得的最大学分,x的子结点集合为son[x],子结点个数为p,且对于x的第i个子结点son[i],以其为根结点的子树中选课数量为$C_i$,则转移方程为:$$f[x][t]=max(\sum_{i=1}^{p}f[son[i]][c[i]])+s[i](满足\sum_

Luogu P2014 选课 (树形DP)

题目 题目描述 在大学里每个学生,为了达到一定的学分,必须从很多课程里选择一些课程来学习,在课程里有些课程必须在某些课程之前学习,如高等数学总是在其它课程之前学习.现在有N门功课,每门课有个学分,每门课有一门或没有直接先修课(若课程a是课程b的先修课即只有学完了课程a,才能学习课程b).一个学生要从这些课程里选择M门课程学习,问他能获得的最大学分是多少? 输入输出格式 输入格式: 第一行有两个整数N,M用空格隔开.(1<=N<=300,1<=M<=300) 接下来的N行,第i+1行

【题解】Luogu P2014 选课

Problem 树上背包问题的典例,记下来 solution 设\(dp[x][t]\)表示以\(x\)为子树,选\(t\)门课获得的最大学分 设\(p\)是\(x\)的子节点数量,\(c_i\)是\(x\)的子节点\(y_i\)选修的课数 转移方程如下 \[dp[x][t]=max_{\sum_{i=1}^pc_i=t-1}\{\sum_{i=1}^pdp[y_i][c_i]\}+pnt[x]\] 事实上,这是一个分组背包模型,对于每个节点\(x\),每个子节点\(y_i\)是一个组,在其中选

LuoGu P2014选课(人生第一个树上背包)

(著名哲学家沃兹基硕德曾经说过:“$QuickSilverX$ $is$ $a$ $BB$”) 就是课与课可能有一些优先关系 这种关系我们可以通过图论建模来解决 不难发现,若将优先选修课向当前课连边,就会生成森林(每门课只有一个选修课,也就只有一条入边) 将所有无入边(没有优先课)的结点与0相连,形成树 ~~不难~~发现这是一个树上DP与背包... 树本身就是个递归的结构,我们每个结点的状态肯定是先递归处理儿子结点的状况来转移的 定义状态 $F[i][j][k]$ 表示第 $i$ 的前 $j$

树形DP 洛谷P2014 选课

P2014 选课 题目描述 在大学里每个学生,为了达到一定的学分,必须从很多课程里选择一些课程来学习,在课程里有些课程必须在某些课程之前学习,如高等数学总是在其它课程之前学习.现在有N门功课,每门课有个学分,每门课有一门或没有直接先修课(若课程a是课程b的先修课即只有学完了课程a,才能学习课程b).一个学生要从这些课程里选择M门课程学习,问他能获得的最大学分是多少? 输入输出格式 输入格式: 第一行有两个整数N,M用空格隔开.(1<=N<=300,1<=M<=300) 接下来的N行

P2014 选课 - 树形DP[树形背包DP]

P2014 选课 传送门 思路: 树形背包DP模型,\(f[i,j]\)表示以\(i\)为根的子树中,选了\(j\)门课的最大学分.树形DP常以子树\(i\)为阶段.树形背包DP相当于树上分组背包DP.\(f[u,j]=max\{f[u,j],f[u,j-k]+f[v,k]~|~v\in~son(u)\}\).我们枚举从u的子树v中选的课数k,将\(f[v,k]\)作为获得的价值加到\(f[u,j-k]\)得到\(f[u,j]\).注意到当前子树根节点u是必须选的,所以要从\(f[u,j-1]\

P2014 选课 题解(树形DP)

题目链接 P2014 选课 解题思路 树形动归,用\(f[i][j]\)表示以\(i\)为根,\(j\)个子节点(不包括自己)的最大学分 首先根据题意建图,用根节点\(0\)将森林连成树. 从根节点开始\(DFS\)遍历,遍历到叶节点后回溯,回溯过程中将\(f[i][j]\)更新,利用背包的思想. \(DFS\)过程中,\(num\)为离根节点0更近的定点,遍历的\(i\)为\(num\)的子节点,容易得出递推关系式: \(f[num][j]=max\{f[num][j],f[num][j-k-

结题报告:luogu P2014

题目链接:P2014 选课 简单的树形\(dp\),借助\(dfs\)实现. 一般的树形\(dp\)数组是需要二维的,其中一维记录节点(编号或父/子节点的状态(有时三维)),另一维记录权值或计数. 重要的是判断从根节点\(dp\)还是从叶节点\(dp\),显然此题需从叶节点开始. 我们记\(dp[i][j]\)为从\(i\)节点向下选\(j\)个节点最大权值(注意不包括自己),易得方程: \[dp[i][j]=max(dp[i][j],dp[i][j-k-1]+dp[i_{son}][k]\]