T1
【问题描述】
一棵n个点的树,你需要给每条边定向。问有多少种方案,使得恰好
有m个点没有入度。
答案对109 + 7取模。
【输入格式】
从文件 a.in 中读入数据。
第一行两个整数n, m。
接下来n − 1行,每行两个数表示一条边。
【输出格式】
输出到文件 a.out 中。
一行一个整数表示答案。
【样例输入】
4 2
1 2
2 3
2 4
【样例输出】
3
【数据规模】
40%的数据,m ≤ n ≤ 20。
60%的数据,m ≤ n ≤ 100。
另有20%的数据,m = 0。
对于100%的数据,m ≤ n ≤ 5000。
思路
- 树形背包,设计状态f[u][sum][0/1]表示以u为根,sum个无初度,u这个根是否有初度。
- 转移分情况讨论,每次转移第二维从0到siz枚举,复杂度O(N2)
#include<bits/stdc++.h>
using namespace std;
const int N=5010;
const int MOD=1e9+7;
int dp[N][N][2],siz[N],head[N],n,m,x,y,cnt,f[N][N];//f[][]防止自己更新自己
struct Edge
{
int to,nxt;
}e[N*2];
void add(int x,int y)
{
e[++cnt].to=y;
e[cnt].nxt=