为什么只有一个黑点的时候还能对答案产生贡献
查看原帖
为什么只有一个黑点的时候还能对答案产生贡献
111349
bobzbh楼主2023/2/20 15:22
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#define N 2086
using namespace std;
int n,k,head[N],tot=0,siz[N],dp[N][N];
struct node
{
	int to,nxt,val;
}e[N];
void adde(int from,int to,int val)
{
	e[++tot].nxt=head[from];
	head[from]=tot;
	e[tot].to=to;
	e[tot].val=val;
}
void dfs(int p,int fa)
{
	siz[p]++;
	for(int i=head[p]; i; i=e[i].nxt)
	{
		int val=e[i].val,to=e[i].to;
		if(to!=fa)
		{
			dfs(to,p);
			siz[p]+=siz[to];
		}
	}
	for(int i=min(siz[p],k); i>=0; --i)
	{
		for(int j=head[p]; j; j=e[j].nxt)
		{
			int to=e[j].to,val=e[j].val;
			if(to!=fa)
			{
				for(int l=0; l<=i&&l<=siz[to]; ++l)
				{
					int black=l,white=siz[to]-l,black_=k-black,white_=n-k-white;
					dp[p][i]=max(dp[p][i],dp[p][i-l]+dp[to][l]+val*(black*black_+white*white_));
				}
			}
			
		}
	}
}
int main()
{
	cin.tie(0),cout.tie(0);
	ios::sync_with_stdio(false);
	cin>>n>>k;
	for(int i=1; i<n; ++i)
	{
		int a,b,val;
		cin>>a>>b>>val;
		adde(a,b,val);
		adde(b,a,val);
	}
	dfs(1,0);
	for(int i=1; i<=n; ++i)
	{
		for(int j=0; j<=k; ++j)
		{
			cout<<dp[i][j]<<" ";
		}
		cout<<endl;
	}
	return 0;
}
/*
错误样例
6 1
2 1 1
3 2 1
4 3 1
5 2 1
6 1 1
7 3 4
8 4 5
9 7 8
*/
/*
dp数组正确答案
22 24   <-这里
14 14
4 4
0 0
0 0
0 0
*/
//dp[1][1]>dp[1][0]?不成对的黑点按理说不应该产生贡献的吧,对拍的时候发现的这个问题。我转移出来dp[1]是22 22
2023/2/20 15:22
加载中...