treecnt

treecnt

﹡    LH (命题人)

基准时间限制:1 秒 空间限制:131072 KB 分值: 40

给定一棵n个节点的树,从1到n标号。选择k个点,你需要选择一些边使得这k个点通过选择的边联通,目标是使得选择的边数最少。

现需要计算对于所有选择k个点的情况最小选择边数的总和为多少。

样例解释:

一共有三种可能:(下列配图蓝色点表示选择的点,红色边表示最优方案中的边)

选择点{1,2}:至少要选择第一条边使得1和2联通。

 

选择点{1,3}:至少要选择第二条边使得1和3联通。

选择点{2,3}:两条边都要选择才能使2和3联通。

Input

第一行两个数n,k(1<=k<=n<=100000)
接下来n-1行,每行两个数x,y描述一条边(1<=x,y<=n)

Output

一个数,答案对1,000,000,007取模。

Input示例

3 2
1 2
1 3

Output示例

4分析:对每条边算贡献即可;   当这条边有贡献时,k个点必然分布在这条边分隔开的两部分中,这里考虑用组合数计算情况数。   合法情况数=总情况数-不合法情况数。   总情况数等于C(n,k),设其中一部分点数为x,另一部分则为n-x,不合法情况数等于C(x,k)+C(n-x,k)。代码:
#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <algorithm>
#include <climits>
#include <cstring>
#include <string>
#include <set>
#include <map>
#include <unordered_map>
#include <queue>
#include <stack>
#include <vector>
#include <list>
#define rep(i,m,n) for(i=m;i<=n;i++)
#define rsp(it,s) for(set<int>::iterator it=s.begin();it!=s.end();it++)
#define mod 1000000007
#define inf 0x3f3f3f3f
#define vi vector<int>
#define pb push_back
#define mp make_pair
#define fi first
#define se second
#define ll long long
#define pi acos(-1.0)
#define pii pair<int,int>
#define Lson L, mid, ls[rt]
#define Rson mid+1, R, rs[rt]
#define sys system("pause")
#define intxt freopen("in.txt","r",stdin)
const int maxn=1e5+10;
using namespace std;
int gcd(int p,int q){return q==0?p:gcd(q,p%q);}
ll qpow(ll p,ll q){ll f=1;while(q){if(q&1)f=f*p%mod;p=p*p%mod;q>>=1;}return f;}
inline ll read()
{
    ll x=0;int f=1;char ch=getchar();
    while(ch<‘0‘||ch>‘9‘){if(ch==‘-‘)f=-1;ch=getchar();}
    while(ch>=‘0‘&&ch<=‘9‘){x=x*10+ch-‘0‘;ch=getchar();}
    return x*f;
}
int n,m,k,t,son[maxn];
ll fac[maxn],c[maxn],ans;
vi e[maxn];
void dfs(int now,int pre)
{
    son[now]=1;
    for(int x:e[now])
    {
        if(x==pre)continue;
        dfs(x,now);
        son[now]+=son[x];
    }
    ans+=c[n]-c[son[now]]-c[n-son[now]];
    ans=(ans+2*mod)%mod;
}
void init()
{
    fac[0]=1;
    for(int i=1;i<=maxn-10;i++)fac[i]=fac[i-1]*i%mod;
    for(int i=m;i<=maxn-10;i++)c[i]=fac[i]*qpow(fac[i-m],mod-2)%mod*qpow(fac[m],mod-2)%mod;
}
int main()
{
    int i,j;
    scanf("%d%d",&n,&m);
    rep(i,1,n-1)scanf("%d%d",&j,&k),e[j].pb(k),e[k].pb(j);
    init();
    dfs(1,0);
    printf("%lld\n",ans);
    //system("Pause");
    return 0;
}
时间: 2024-10-17 10:20:47

treecnt的相关文章

treecnt 算法马拉松20(告别美国大选及卡斯特罗)

treecnt 基准时间限制:1 秒 空间限制:131072 KB 给定一棵n个节点的树,从1到n标号.选择k个点,你需要选择一些边使得这k个点通过选择的边联通,目标是使得选择的边数最少. 现需要计算对于所有选择k个点的情况最小选择边数的总和为多少. 样例解释: 一共有三种可能:(下列配图蓝色点表示选择的点,红色边表示最优方案中的边) 选择点{1,2}:至少要选择第一条边使得1和2联通.   选择点{1,3}:至少要选择第二条边使得1和3联通. 选择点{2,3}:两条边都要选择才能使2和3联通.

51Nod 1677 treecnt 【树形dp+组合数学+逆元】

51Nod 1677  treecnt Description:给定一棵n个节点的树,从1到n标号.选择k个点,你需要选择一些边使得这k个点通过选择的边联通,目标是使得选择的边数最少. 现需要计算对于所有选择k个点的情况最小选择边数的总和为多少. 样例解释: 一共有三种可能:(下列配图蓝色点表示选择的点,红色边表示最优方案中的边) 选择点{1,2}:至少要选择第一条边使得1和2联通.   选择点{1,3}:至少要选择第二条边使得1和3联通. 选择点{2,3}:两条边都要选择才能使2和3联通. I

树上统计treecnt(dsu on tree 并查集 正难则反)

题目链接 \(Description\) 给定一棵\(n(n\leq 10^5)\)个点的树. 定义\(Tree[L,R]\)表示为了使得\(L\sim R\)号点两两连通,最少需要选择的边的数量. 求\[\sum_{l=1}^n\sum_{r=l}^nTree[l,r]\] \(Solution\) 枚举每条边,计算它的贡献. 那么我们要判断有多少连续区间的点跨过这条边,并不好算,反过来去求在这条边的两侧分别有多少个连续区间. 那么显然有\(O(n^2)\)的做法,即对每条边DFS它的两侧,枚

51nod_1677:treecnt

题目是求一棵n节点树中对于C(n,k)颗子树,每棵子树为在n个节点中选不同的k个节点作为树的边界点,这样的所有子树共包含多少条边. 问题可以转化一下,对每一条边,不同的子树中可能包含可能不包含这条边,显然,只有子树那k个节点在该边的两侧均有分布时该边才被包含在子树中.所有边的被包含次数的和,即为answer.对于一条边的被包含次数,设该边两侧分别有a,b个节点,那么,该边被包含的次数为C(a+b,k)-C(a,k)-C(b,k)(也可以借助母函数函数求C(a,i)*C(b,k-i),i从1到mi

51Nod 1677 treecnt

一道比较基础的计数题,还是一个常用的单独计算贡献的例子. 首先看题目和范围,暴力枚举肯定是不可行的,而且\(O(n\ logn)\)的算法貌似很难写. 那我们就来想\(O(n)\)的吧,我们单独考虑每一条边的贡献,我们注意到一个重要的性质: 树上任意两点间的最短路径都是唯一确定的. 这个常识吧,所以我们只需要考虑每一条边两边的点在计算时会经过这条边多少次. 我们枚举每一条边,然后可以这样考虑这一条边: 我们设一边有\(x\)个点,另一边有\(y\)个点,很明显\(x+y=n\) 然后我们考虑有多

UVA 12219-Common Subexpression Elimination

原题链接:点击此处 觉得这题好难呀!方法在紫书上,十一节公共表达式消除那小节(没带书回来,具体页数不清楚啦~) 这题树中的结点对应的子树都是二叉树,不存在只有一个儿子的情况.输入的字符串中也只有两种情况,结点名字后面紧跟一个左括号的,和不跟括号的. 因此很容易判断一个结点是否为叶子结点:只要在原字符串中看他后面有没有紧跟着左括号就行了.如果有,立刻递归建树作为左子树.因为有左子树就一定有右子树,所以左子树建完后就可以从字符串中左子树结束的位置开始建右子树了.因此不需要从左到右扫描寻找逗号来分割左

BZOJ 1036: [ZJOI2008]树的统计Count (树链剖分模板题)

1036: [ZJOI2008]树的统计Count Time Limit: 10 Sec  Memory Limit: 162 MBSubmit: 14982  Solved: 6081[Submit][Status][Discuss] Description 一棵树上有n个节点,编号分别为1到n,每个节点都有一个权值w.我们将以下面的形式来要求你对这棵树完成一些操作: I. CHANGE u t : 把结点u的权值改为t II. QMAX u v: 询问从点u到点v的路径上的节点的最大权值 I

SBT模板

1 struct SBT{ 2 int treeCnt, root; 3 struct tnode{int data, lc, rc, size;} tree[maxn]; 4 inline void clear(){treeCnt = 0; root = 0;} 5 inline void leftRot(int &x){ 6 int k = tree[x].rc; 7 tree[x].rc = tree[k].lc; 8 tree[k].lc = x; 9 tree[k].size = tr

树的计数

题目: 1.树的计数 [问题描述] 图和树有很密切的关系.某一天牙神产生了一个很奇怪的想法:删去一些边把一个无向图变成一个树,也就是将边留下 N ? 1 条.而且对于任意一个点 i,要保证现在树中的边满足 1 号点到 i 号点的路径长度等于原图中 1 号点到 i 号点的最短路长度.牙神请你帮忙数一下有多少个目标树满足要求.答案对1000000007 取模. [输入格式] 第一行,一个整数N.接下来 N 行每行 N 个数,第 i 行第 j 个数表示 i 和 j 的连边关系,0 表示没有边. [输出