Codeforces Round #656 (Div. 3) F. Removing Leaves 贪心 + 模拟

最新推荐文章于 2023-09-01 22:24:26 发布

YB Lin

最新推荐文章于 2023-09-01 22:24:26 发布

阅读量64

点赞数

分类专栏：贪心

本文链接：https://blog.csdn.net/m0_51068403/article/details/117387949

版权

贪心算法图论删除叶子节点集合操作 C++

关键词由CSDN通过智能技术生成

贪心专栏收录该内容

16 篇文章 0 订阅

订阅专栏

传送门

文章目录

题意：
思路：

题意：

在这里插入图片描述

思路：

首先有一个贪心策略就是每次都找一个叶子节点最多的点，让后删掉他的 $k$ 个叶子节点，现在我们就来考虑如何模拟这个过程。
我们整一个 $v e c t o r < s e t < i n t > >$ 来存边和叶子，方便删除点，让后再整一个 $s e t$ 按照 $l e a f [i] . s i z e ()$ 排序，由于是按照可变大小的值来排序的，那么必须要先从 $s e t$ 中删除这个点，让后再修改 $l e a f$ ，再加入这个点才可以，不然出大问题。之后我们就可以模拟删除点了，要提前预处理出来 $l e a f$ ，而且删除点的时候有肯能某个点也成为叶子节点，这个时候也需要修改 $l e a f$ ，注意细节就好。

由于我nt了，先该的值再删点，代码已经面目全非，跟题解差不多了。这个题主要还是要找到合适的方法去模拟删边就比较好做辣。

最后还要注意特判 $k = = 1$ 的情况。

// Problem: F. Removing Leaves
// Contest: Codeforces - Codeforces Round #656 (Div. 3)
// URL: https://codeforces.com/contest/1385/problem/F
// Memory Limit: 256 MB
// Time Limit: 2000 ms
// 
// Powered by CP Editor (https://cpeditor.org)

//#pragma GCC optimize("Ofast,no-stack-protector,unroll-loops,fast-math")
//#pragma GCC target("sse,sse2,sse3,ssse3,sse4.1,sse4.2,avx,avx2,popcnt,tune=native")
//#pragma GCC optimize(2)
#include<cstdio>
#include<iostream>
#include<string>
#include<cstring>
#include<map>
#include<cmath>
#include<cctype>
#include<vector>
#include<set>
#include<queue>
#include<algorithm>
#include<sstream>
#include<ctime>
#include<cstdlib>
#define X first
#define Y second
#define L (u<<1)
#define R (u<<1|1)
#define pb push_back
#define mk make_pair
#define Mid (tr[u].l+tr[u].r>>1)
#define Len(u) (tr[u].r-tr[u].l+1)
#define random(a,b) ((a)+rand()%((b)-(a)+1))
#define db puts("---")
using namespace std;

//void rd_cre() { freopen("d://dp//data.txt","w",stdout); srand(time(NULL)); }
//void rd_ac() { freopen("d://dp//data.txt","r",stdin); freopen("d://dp//AC.txt","w",stdout); }
//void rd_wa() { freopen("d://dp//data.txt","r",stdin); freopen("d://dp//WA.txt","w",stdout); }

typedef long long LL;
typedef unsigned long long ULL;
typedef pair<int,int> PII;

const int N=1000010,mod=1e9+7,INF=0x3f3f3f3f;
const double eps=1e-6;

int n,k;
vector<set<int>>v,leaf;

struct cmp {
	bool operator () (int a,int b) const {
		if(leaf[a].size()==leaf[b].size()) return a<b;
		return leaf[a].size()>leaf[b].size();
	}
};

int solve() {
	if(k==1) return n-1;
	set<int,cmp>s;
	int ans=0;
	for(int i=0;i<n;i++) s.insert(i);
	while(1) {
		int now=*s.begin();
		if(leaf[now].size()<k) break;
		for(int i=1;i<=k;i++) {
			int to=*leaf[now].begin(); 
			
			s.erase(now); s.erase(to);
			
			leaf[now].erase(to);
			if(leaf[to].count(now)) leaf[to].erase(now);
			
			v[now].erase(to); v[to].erase(now);
			if(v[now].size()==1) {
				int ne=*v[now].begin();
				s.erase(ne); 
				leaf[ne].insert(now);
				s.insert(ne);
			}
			
			s.insert(now); s.insert(to);
		}
		ans++;
	}
	return ans;
}

int main()
{
//	ios::sync_with_stdio(false);
//	cin.tie(0);
	
	int _; scanf("%d",&_);
	while(_--) {
		scanf("%d%d",&n,&k);
		v=vector<set<int>>(n);
		leaf=vector<set<int>>(n);
		for(int i=1;i<=n-1;i++) {
			int a,b; scanf("%d%d",&a,&b);
			a--; b--;
			v[a].insert(b); v[b].insert(a);
		}
		for(int i=0;i<n;i++) 
			if(v[i].size()==1) {
				leaf[*v[i].begin()].insert(i);
			}
		printf("%d\n",solve());
	}



	return 0;
}
/*

*/