Problem - G - Codeforces
题意:
给定一棵n个节点的树,请你减掉一些边,使得剪掉后的每个树只有三个节点,
如果可以,第一行返回减掉边的数量,第二行返回减掉边的编号;如果无解,输出-1
思路:
首先去观察样例:
可以发现我们可以从叶子结点递归裁剪
Code:
#include <bits/stdc++.h>#define int long longusing 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;
}