VK Cup 2016 - Qualification Round 2 C. Road Improvement dfs

题意

给你一棵树，n点n-1条边，现在这些边都不存在，你要修起来。

你可以选择一条边修理，如果这个边的两边的城市都没有在修理其他边的话。

每次修理需要一天的时间。

问你最少多少天可以修完，并且把方案输出。

题解：

直接dfs就好了，我们记录一个上一个边传过来的时间和这条边修理的时间就好了

保证不重复就行了。

#include <set>
#include <map>
#include <stack>
#include <queue>
#include <deque>
#include <cmath>
#include <vector>
#include <string>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
#define L(i) i<<1
#define R(i) i<<1|1
#define INF  0x3f3f3f3f
#define pi acos(-1.0)
#define eps 1e-12
#define maxn 200100
#define MOD 1000000007

int n;
vector<pair<int,int> > E[maxn];
vector<int> ans[maxn];
int cnt;
void dfs(int u,int pre,int time)
{
int now = 0;
for(int i = 0; i < E[u].size(); i++)
{
int v = E[u][i].first;
if(v == pre)
continue;
now++;
if(now == time)
now++;
cnt = max(cnt,now);
ans[now].push_back(E[u][i].second);
dfs(v,u,now);
}
}
int main()
{
int t;
//scanf("%d",&t);
while(scanf("%d",&n) != EOF)
{
for(int i = 1; i < n; i++)
{
int x,y;
scanf("%d%d",&x,&y);
E[x].push_back(make_pair(y,i));
E[y].push_back(make_pair(x,i));
ans[i].clear();
}
cnt = 0;
dfs(1,-1,0);
printf("%d\n",cnt);
for(int i = 1; i <= cnt; i++)
{
printf("%d ",ans[i].size());
for(int j = 0; j < ans[i].size(); j++)
printf("%d ",ans[i][j]);
printf("\n");
}
}
return 0;
}