-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathsibling dp
110 lines (92 loc) · 1.67 KB
/
sibling dp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
using namespace std;
#define ma 402
ll cld[ma],sib[ma],dp[ma][ma],k,pa[ma],path[ma][ma];
vector<ll>adj[ma],ans;
void dfs(ll u,ll par)
{
ll x=0;
ll i;
f(i,0,adj[u].sz)
{
ll p=adj[u][i];
if(p==par)continue;
pa[p]=u;
if(sib[x]==0)sib[x]=p;
if(cld[u]==0)cld[u]=p;
dfs(p,u);
x=p;
}
}
ll cal(ll u,ll k)
{
if(u==0)
{
if(k>0)return inf;
return 0;
}
ll i;
ll &r=dp[u][k];
if(r!=-1)return r;
r=1+cal(sib[u],k);
for(i=0;i<k;i++)
{
r=min(r,cal(cld[u],i)+cal(sib[u],k-1-i));
}
return r;
}
void pat(ll u,ll k,ll rem)
{
ll i;
if(rem==0)return;
if(dp[u][k]==1+cal(sib[u],k))
{
ll p=pa[u];
ans.pb(path[u][p]);
pat(sib[u],k,rem-1);
return;
}
for(i=0;i<k;i++)
{
ll p=cal(cld[u],i)+cal(sib[u],k-i-1);
if(p==dp[u][k])
{
pat(cld[u],i,rem);
pat(sib[u],k-i-1,rem);
return;
}
}
}
int main()
{
ll n,u,v,i;
while(cin>>n>>k)
{
f(i,1,n)
{
cin>>u>>v;
adj[u].pb(v);
adj[v].pb(u);
path[u][v]=path[v][u]=i;
}
dfs(1,-1);
memset(dp,-1,sizeof(dp));
ll min1=inf;
ll node=-1;
for(i=1;i<=n;i++)
{
ll r=cal(cld[i],k-1)+(i!=1);
if(min1>r)node=i;
min1=min(min1,r);
}
ll p=pa[node];
if(p>0)
{
ans.pb(path[p][node]);
}
pat(cld[node],k-1,min1);
cout<<min1<<endl;
f(i,0,ans.sz)cout<<ans[i]<<" ";
cout<<endl;
}
return 0;
}