原题链接:年会
解题思路:
记忆化搜索 + 二叉树
常理来说大学里 一个导师可以有多个学员 但一个学员只能有一个导师(二叉树)
导师和学员不能同时邀请,那就是除了同时邀请外 还有三种情况,邀导不邀学
邀学不邀导 和都不邀 (0与1 )的关系 通过DFS枚举各类情况,优化记忆数组;
#include <string.h> #include <stdio.h> #define M 6002 #define max(a,b) a>b?a:b int n; int v[M]; int bj[M]; //标记非根节点(将有学员身份的员工或者学员标记) //没标记的就只有导师身份(即根节点) int ds[M][5001]; //导师数组 ds[i][j]=4;代表第i个人作为导师身份时第j个学员为4号学员 int dp[M][2];//记忆数组 dp[L][0]代表满足条件下不邀请该员工(L)的最优活跃度 // dp[L][1]代表满足条件下邀请该员工(L)的最优活跃度 int dfs(int L,int last) { //优化 记忆数组 if(last==1&&dp[L][0]!=-1)return dp[L][0]; if(last==0&&dp[L][1]!=-1)return dp[L][1]; if(!ds[L][0])//表示 当前员工无学员 即 该员工只有学员身份 { if(last==0&&v[L]>0)return v[L];//若该员工的导师未邀请且该学员可i活跃气氛 else return 0;//不活跃当然不能要; } int ans1=0, ans2=0,i; //若不邀请该员工 就去找该员工的学员(下一层节点)的最优解 i=0;while(ds[L][i]!=0)ans1+=dfs(ds[L][i],0),i++; //若该员工的导师没被邀请 并且邀请该员工 然后去找该员工的学员(下一层节点)的最优解 if(last!=1)i=0;while(ds[L][i]!=0)ans2+=dfs(ds[L][i],1),i++; if(last==0)//若该员工的导师未邀请 {dp[L][1]=max(ans1,ans2+v[L]); //该节点的最优解为(邀请该员工 与不邀请的该员工的)最大值 //(即可能 该员工和他导师活跃值为负 都不邀请滴好) return dp[L][1]; }//若该员工的导师邀请 ,那该员工只能不邀请 dp[L][0]=ans1; return dp[L][0]; } int main() { int i,l,k,j,ans=0; scanf("%d",&n); memset(bj,0, sizeof(bj)); memset(ds,0, sizeof(ds)); memset(dp,-1,sizeof(dp)); for(i=1;i<=n;i++) scanf("%d",&v[i]); while(1) { scanf("%d%d",&l,&k); if(l==0&&k==0)break; j=0; while(ds[k][j]!=0)j++;//给l学员在他导师那找个序号位置 ds[k][j]=l; bj[l]=1;//标记该员工有学员身份 } for(i=1;i<=n;i++)//枚举根节点( 只有导师身份的人) if(!bj[i])ans+=dfs(i,0);//把各各没有关系的导师集合的最优解加起来 printf("%d\n",ans); return 0; }
0.0分
4 人评分
C语言网提供由在职研发工程师或ACM蓝桥杯竞赛优秀选手录制的视频教程,并配有习题和答疑,点击了解:
一点编程也不会写的:零基础C语言学练课程
解决困扰你多年的C语言疑难杂症特性的C语言进阶课程
从零到写出一个爬虫的Python编程课程
只会语法写不出代码?手把手带你写100个编程真题的编程百练课程
信息学奥赛或C++选手的 必学C++课程
蓝桥杯ACM、信息学奥赛的必学课程:算法竞赛课入门课程
手把手讲解近五年真题的蓝桥杯辅导课程
发表评论 取消回复