求助40pts(WA #2 #5 #7 #8 #9 #10)
查看原帖
求助40pts(WA #2 #5 #7 #8 #9 #10)
375953
Lgx_Q楼主2022/11/30 18:58
#include<bits/stdc++.h>
using namespace std;
const int maxn=1010;
int n,m,ans,d[maxn],w[maxn],v[maxn],W[maxn],V[maxn],dfn[maxn],low[maxn],T,in[maxn],head[maxn],tot,stk[maxn],top,cnt,c[maxn],f[maxn][maxn];
struct edge
{
	int v,nxt;
}e[maxn];
void insert(int u,int v)
{
	e[++tot]=(edge){v,head[u]};
	head[u]=tot;
}
void tarjan(int u)
{
	dfn[u]=low[u]=++T;
	stk[++top]=u;
	for(int i=head[u];i;i=e[i].nxt)
	{
		int v=e[i].v;
		if(dfn[v])
		{
			low[u]=min(low[u],dfn[v]);
		}
		else
		{
			tarjan(v);
			low[u]=min(low[u],low[v]);
		}
	}
	if(dfn[u]==low[u])
	{
		++cnt;
		while(stk[top+1]!=u)
		{
			W[cnt]+=w[stk[top]];
			V[cnt]+=v[stk[top]];
			c[stk[top]]=cnt;
			--top;
		}
	}
}
void dfs(int u)
{
	for(int i=head[u];i;i=e[i].nxt)
	{
		int v=e[i].v;
		dfs(v);
	}
	f[u][W[u]]=V[u];
	for(int i=head[u];i;i=e[i].nxt)
	{
		int v=e[i].v;
		for(int j=m;j>=W[u];j--)
		{
			for(int k=0;k+W[u]<=j;k++)
			{
				f[u][j]=max(f[u][j],f[u][j-k]+f[v][k]);
			}
		}
	}
//	for(int j=0;j<=m;j++)
//	{
//		printf("f[%d][%d] = %d\n",u,j,f[u][j]);
//	}
}
int main()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++) scanf("%d",&w[i]);
	for(int i=1;i<=n;i++) scanf("%d",&v[i]);
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&d[i]);
		if(d[i]) insert(d[i],i);
	}
	for(int i=1;i<=n;i++)
	{
		if(!dfn[i])
		{
			tarjan(i);
		}
	}
	tot=0;
	memset(head,0,sizeof head);
	for(int i=1;i<=n;i++)
	{
		if(d[i]&&c[d[i]]!=c[i])
		{
			insert(c[d[i]],c[i]);
			++in[c[i]];
		}
	}
	for(int i=1;i<=cnt;i++)
		if(!in[i])
		{
			insert(0,i);
		}
	memset(f,0xcf,sizeof f);
	dfs(0);
	for(int i=0;i<=m;i++) ans=max(ans,f[0][i]);
	printf("%d",ans);
	return 0;
}
2022/11/30 18:58
加载中...