题目链接
题意:一个树形网络,编号为1的是广播站,叶子节点为广播接收者,要想使用必须付出费用,中间的其他点是中继站,信号到达每个中继站都会有一定的消费;问在保证广播站收益 不亏本的情况下使用者最多是多少?
分析:dp[i][j]表示第i个点下面有j个使用用户的收益;状态转移方程: dp[u][j]=max(dp[u][j],dp[u][j-k]+dp[v][k]-edge[i].w); 其中u是v的父节点,j枚举u节点的所有子叶点数,k枚举v节点的所有子叶点数,edge[i].w表示u节点下的使用用户不是0的话要减去的成本 dp全部初始化为-inf,dp[i][0]赋为0;
#include<cstdio>
#include<cstring>
#include<vector>
#include<algorithm>
#define INF 0x3f3f3f3f
using namespace std;
int dp[3300][3300];
//dp[i][j]表示第i个点下面有j个使用用户的收益;状态转移方程
//dp[u][j]=max(dp[u][j],dp[u][j-k]+dp[v][k]-edge[i].w)
//u是v的父节点,j枚举u节点的所有子叶点数,k枚举v节点的所有点数
//edge[i].w表示u节点下的使用用户不是0的话要减去的成本
struct node
{
int v,w;
node(int vv,int ww)//学学
{
v=vv;
w=ww;
}
};
vector<node>edge[3300];
int cost[3300],sum[3300],use[3300],m,n;
void dfs(int u)
{
use[u]=1;//标记已经用过
for(int i=0;i<(int)edge[u].size();i++)
{
int v=edge[u][i].v;//u的子节点v
if(!use[v])//v没有跑过
{
dfs(v);//往下搜
sum[u]+=sum[v];//加上子节点用户个数
for(int j=sum[u];j>=0;j--)
for(int k=1;k<=sum[v];k++)
if(j>=k)
dp[u][j]=max(dp[u][j],dp[u][j-k]+dp[v][k]-edge[u][i].w);
}
}
if(u>n-m)//走到潜在用户
{
sum[u]=1;//计数为1
dp[u][1]=cost[u];//1个潜在用户的收益
}
}
int main()
{
//freopen("in.txt","r",stdin);
int i,j,k,c;
while(scanf("%d%d",&n,&m)!=-1)
{
for(i=1;i<=n;i++)
edge[i].clear();
for(i=1;i<=n-m;i++)
{
scanf("%d",&k);
while(k--)
{
scanf("%d%d",&j,&c);//号码,费用
edge[i].push_back(node(j,c));
edge[j].push_back(node(i,c));
}
}
for(i=n-m+1;i<=n;i++)
scanf("%d",&cost[i]);//支付价格
for(i=1;i<=n;i++)
{
dp[i][0]=0;
for(j=1;j<=m;j++)
dp[i][j]=-INF;
}
memset(sum,0,sizeof(sum));
memset(use,0,sizeof(use));
dfs(1);//发射机
for(i=m;i>=1;i--)
if(dp[1][i]>=0)
{
printf("%d\n",i);
break;
}
}
return 0;
}