Problem - G - Codeforces
题意:
给定一棵n个节点的树,请你减掉一些边,使得剪掉后的每个树只有三个节点,
如果可以,第一行返回减掉边的数量,第二行返回减掉边的编号;如果无解,输出-1
思路:
首先去观察样例:
可以发现我们可以从叶子结点递归裁剪
Code:
#include <bits/stdc++.h>
#define int long long
using namespace std;
const int mxn=2e5+10;
const int mxe=2e5+10;
struct ty{
int to,next;
}edge[mxe<<2];
vector<int> ans;
map<pair<int,int>,int> G;
int N,u,v;
int tot=0;
int head[mxn];
void add(int u,int v){
edge[tot].to=v;
edge[tot].next=head[u];
head[u]=tot++;
}
void G_init(){
tot=0;
G.clear();
ans.clear();
for(int i=0;i<=N;i++){
head[i]=-1;
}
}
int dfs(int u,int fa){
int cnt=1;
for(int i=head[u];~i;i=edge[i].next){
if(edge[i].to==fa) continue;
cnt+=dfs(edge[i].to,u);
}
if(cnt==3){
ans.push_back(G[{u,fa}]);
return 0;
}
return cnt;
}
void solve(){
cin>>N;
G_init();
for(int i=1;i<=N-1;i++){
cin>>u>>v;
add(u,v);
add(v,u);
G[{u,v}]=G[{v,u}]=i;
}
if(N%3!=0){
cout<<-1<<'\n';
return;
}
int x=dfs(1,0);
if(ans.size()!=N/3){
cout<<-1<<'\n';
return;
}
cout<<N/3-1<<'\n';
for(int i=0;i<ans.size()-1;i++) cout<<ans[i]<<" \n"[i==ans.size()-2];
}
signed main(){
ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
int __=1;cin>>__;
while(__--)solve();return 0;
}