这题为什么不能贪心?
  • 板块P2015 二叉苹果树
  • 楼主anda
  • 当前回复15
  • 已保存回复15
  • 发布时间2020/7/18 15:47
  • 上次更新2023/11/6 22:53:42
查看原帖
这题为什么不能贪心?
157571
anda楼主2020/7/18 15:47
#include<iostream>
#include<queue>
#include<vector>
using namespace std;
int n,q,in[101],ans;
struct node{
	int to,data;
	bool operator <(const node &x)const{
		return x.data<data;
	}
}fa[101];
vector<node>g1[101],g[101];
priority_queue<node>s;
void dfs(int x,int pre){
	for(int i=0;i<g1[x].size();i++){
		if(g1[x][i].to!=pre){
			fa[g1[x][i].to]=(node){x,g1[x][i].data};
			dfs(g1[x][i].to,x);
		}
	}
}
int main(){
	cin>>n>>q;q=n-1-q;
	for(int i=1;i<n;i++){
		int x,y,z;
		cin>>x>>y>>z;ans+=z;
		g1[x].push_back((node){y,z});
		g1[y].push_back((node){x,z});
	}
	dfs(1,0);in[0]=0x7fffffff;
	for(int i=1;i<=n;i++)in[fa[i].to]++;
	for(int i=1;i<=n;i++)if(!in[i])s.push(fa[i]);
	while(q--){
		node x=s.top();ans-=x.data;s.pop();
			in[fa[x.to].to]--; 
			if(!in[fa[x.to].to])s.push(fa[x.to]);
	}
	cout<<ans;
}
2020/7/18 15:47
加载中...