UOJ#351. 新年的叶子 概率期望

原文链接https://www.cnblogs.com/zhouzhendong/p/UOJ351.html

题目传送门 - UOJ351

题意

  有一个 n 个节点的树,每次涂黑一个叶子节点(度为 1 的节点),可以重复涂黑。

  问使得白色部分的直径发生变化的期望涂黑次数。

  $n\leq 5\times 10^5$

题解

  首先考虑什么情况下直径长度会发生改变。

  考虑找到直径的中点,可能在边上。

  对于这个直径相连的每一个子树,分别算出在这个子树中的距离直径中点距离为直径长度的一半的节点个数。

  于是我们就得到了一些集合。那么,直径长度将在 只剩下一个集合有白色节点 的时候发生改变。

  于是,很容易得出菊花图的做法:

$$ ans = \sum_{i=1}^{n-1} \frac {n-1} {n-i}$$

  但是不是菊花图的时候,仍然很棘手。

  设总叶子节点个数为 $l$ ,所有集合的元素个数总和为 $tot$ 。

官方题解的算法三应该比较好理解吧

算法三

我们考虑枚举哪个集合最终剩下了,设当前集合大小为 $k$ 。

设所有集合大小的和为 $n$ 。

我们枚举当其他所有集合都染黑时,当前集合还剩下 $i$ 个没有被染黑,那么这样的情况对答案的贡献为

$$\frac{\binom{k}{i}\times \binom{n-i-1}{n-k-1}\times \binom{n-k}{1}\times(k-i)!\times(n-k-1)!\times (i)!}{n!}\times (\sum_{j=i+1}^{n}\frac{m}{j})$$

线性预处理阶乘,阶乘逆元与逆元的前缀和就可以做到 $O(n)$ 的时间复杂度。

算法四比较神仙

算法四

from WuHongxun

http://wuhongxun.blog.uoj.ac/blog/3345

放链接(逃

  对于算法四,我再补充一句:

考虑任何一个方案里我们染黑到所有集合都变黑了为止,它对答案的贡献是多少呢?如果 x 是这个方案里最后被染黑的集合,那么是倒数第二个被染黑的时间,反之则是最后一个集合被染黑的时间。

  这里,如果 x 不是最后一个被染黑的集合,那么必然全部染黑了。

代码

#include <bits/stdc++.h>
using namespace std;
int read(){
	int x=0;
	char ch=getchar();
	while (!isdigit(ch))
		ch=getchar();
	while (isdigit(ch))
		x=(x<<1)+(x<<3)+(ch^48),ch=getchar();
	return x;
}
const int N=500005,mod=998244353;
int n;
vector <int> e[N],S;
int in[N],depth[N],fa[N];
int vis[N];
void dfs(int x,int pre,int d){
	depth[x]=d,fa[x]=pre;
	for (auto y : e[x])
		if (y!=pre)
			dfs(y,x,d+1);
}
int LCA(int x,int y){
	if (depth[x]<depth[y])
		swap(x,y);
	while (depth[x]>depth[y])
		x=fa[x];
	while (x!=y)
		x=fa[x],y=fa[y];
	return x;
}
int FindFar(int x,int d){
	vis[x]=d;
	int res=x;
	for (auto y : e[x])
		if (!~vis[y]){
			int tmp=FindFar(y,d+1);
			if (vis[tmp]>vis[res])
				res=tmp;
		}
	return res;
}
void Get(int x,int d,int md){
	if (d==md)
		S.back()++;
	vis[x]=1;
	for (auto y : e[x])
		if (!vis[y])
			Get(y,d+1,md);
}
void Get_S(){
	memset(vis,-1,sizeof vis);
	int s=FindFar(1,0);
	memset(vis,-1,sizeof vis);
	int t=FindFar(s,0);
	if (depth[s]<depth[t])
		swap(s,t);
	int d=depth[s]+depth[t]-2*depth[LCA(s,t)];
	memset(vis,0,sizeof vis);
	if (d&1){
		for (int i=d/2;i--;)
			s=fa[s];
		t=fa[s];
		vis[s]=vis[t]=1;
		S.push_back(0),Get(s,0,d/2);
		S.push_back(0),Get(t,0,d/2);
	}
	else {
		for (int i=d/2;i--;)
			s=fa[s];
		vis[s]=1;
		for (auto y : e[s])
			S.push_back(0),Get(y,1,d/2);
	}
}
int Pow(int x,int y){
	int ans=1;
	for (;y;y>>=1,x=1LL*x*x%mod)
		if (y&1)
			ans=1LL*ans*x%mod;
	return ans;
}
int Fac[N],Inv[N],Iv[N],h[N];
void Math_Prework(){
	for (int i=Fac[0]=1;i<=n;i++)
		Fac[i]=1LL*Fac[i-1]*i%mod;
	Inv[n]=Pow(Fac[n],mod-2);
	for (int i=n;i>=1;i--)
		Inv[i-1]=1LL*Inv[i]*i%mod;
	for (int i=1;i<=n;i++)
		Iv[i]=1LL*Inv[i]*Fac[i-1]%mod;
	for (int i=1;i<=n;i++)
		h[i]=(h[i-1]+Iv[i])%mod;
}
int C(int n,int m){
	if (m<0||m>n)
		return 0;
	return 1LL*Fac[n]*Inv[m]%mod*Inv[n-m]%mod;
}
int main(){
	n=read();
	for (int i=1;i<n;i++){
		int a=read(),b=read();
		e[a].push_back(b);
		e[b].push_back(a);
		in[a]++,in[b]++;
	}
	dfs(1,0,0);
	Get_S();
	int l=0;
	for (int i=1;i<=n;i++)
		if (in[i]==1)
			l++;
	Math_Prework();
	int tot=0;
	for (auto y : S)
		tot+=y;
	// 设当前剩余 k 个,总共有 l 个,选出 x 个数:
	// l/k + l/(k-1) + ... + l/(k-x+1)
	// l * (h[k]-h[k-x])
	int ans=0;
	for (auto y : S)
		ans=(1LL*h[tot-y]+ans)%mod;
	ans=(-1LL*((int)S.size()-1)*h[tot]%mod+ans+mod)%mod;
	ans=1LL*ans*l%mod;
	printf("%d",ans);
	return 0;
}

原文地址:https://www.cnblogs.com/zhouzhendong/p/UOJ351.html

时间: 2024-11-09 08:48:18

UOJ#351. 新年的叶子 概率期望的相关文章

hdu 4586 (概率+期望)

http://acm.hdu.edu.cn/showproblem.php?pid=4586 大致题意:有一个骰子有n个面,掷到每一个面的概率是相等的,每一个面上都有相应的钱数.其中当你掷到m个面之一时,你有多掷一次的机会.问最后所得钱数的期望. 思路:设投掷第一次的期望是p,那么第二次的期望是m/n*p,第三次的期望是 (m/n)^2*p......第N次的期望是(m/n)^(N-1)*p. 那么这些期望之和便是答案.之前也是想到这,但不知道如何处理无限的情况.当时脑卡了,这不是赤裸裸的等比数

OI队内测试一【数论概率期望】

版权声明:未经本人允许,擅自转载,一旦发现将严肃处理,情节严重者,将追究法律责任! 测试分数:110 本应分数:160 改完分数:200 T1: 题解:推出了一个初始式子但是n的4分之3次方 忘了合并[实际上是没发现]本来应有60分的,但是忘记开long long 只有30分 因为一些公式不好写出来就直接截图题解吧! T2: 题解:很简单的概率期望,算出每个点被选的概率,然后在上树状数组或者线段树求逆序队,但是我只有80分,为什么解法不行?NO NO 你还是太年轻我只是线段树数据范围开小了!![

uvalive 7331 Hovering Hornet 半平面交+概率期望

题意:一个骰子在一个人正方形内,蜜蜂在任意一个位置可以出现,问看到点数的期望. 思路:半平面交+概率期望 1 #include<cstdio> 2 #include<cstring> 3 #include<algorithm> 4 #include<iostream> 5 #include<cstdlib> 6 #include<string> 7 #include<cmath> 8 #include<vector&

【BZOJ-1419】Red is good 概率期望DP

1419: Red is good Time Limit: 10 Sec  Memory Limit: 64 MBSubmit: 660  Solved: 257[Submit][Status][Discuss] Description 桌面上有R张红牌和B张黑牌,随机打乱顺序后放在桌面上,开始一张一张地翻牌,翻到红牌得到1美元,黑牌则付出1美元.可以随时停止翻牌,在最优策略下平均能得到多少钱. Input 一行输入两个数R,B,其值在0到5000之间 Output 在最优策略下平均能得到多少钱

Lightoj 1027 概率期望

Problem: 概率期望 Analyse: 非常经典的带有递归实现,的方程期望式子. E为期望,z为正数个数,zp为正数平均值,f为负数个数,fp为负数平均值. z?zpn+f?fp+En=E /**********************jibancanyang************************** *Author* :jibancanyang *Created Time* : 五 5/ 6 23:58:46 2016 *File Name* : .cpp **Code**:

hdu4405--Aeroplane chess+概率期望dp

首先推荐一篇很好的如何概率期望问题的入门文章:点击打开链接 昨天比赛的时候面对这道题的第一想法是依照数学期望的定义来做,即依次求出某个点扔i次骰子能到达n点的概率,然后由期望的定义就可以求出答案了.但显然这在程序上是不可能实现的. 今天看了那篇文章后才知道自己的想法是大错特错的;求解这种问题应该采用一种递推的思路,即每次只考虑一次转移后当前状态的期望,然后我们依次考虑每个节点就可以得到一个方程组,然后就只需要求解这个方程组就行了. 当然对于如何求解这个方程组,我们可以采用高斯消元法,当然如果这个

CF148D--Bag of mice+概率期望dp

第一道概率期望dp:) 其实和一般的dp也差不多,只要状态选好就行了. 定义dp[i][j]表示还剩i只白老鼠j只黑老鼠时候公主赢得概率. 则:1.公主选白老鼠,直接赢,概率:i/(i+j) 2.公主选黑老鼠 1)龙选黑老鼠,逃走黑老鼠:概率:j/(i+j)*(j-1)/(i+j-1)*(j-2)/(i+j-2) 2)  龙选黑老鼠,逃走白老鼠:概率:j/(i+j)*(j-1)/(i+j-1)*i/(i+j-2) 3) 龙选白老鼠,这样公主是必输的,不用考虑 然后dp[i][j]等于以上概率之和

浅谈概率期望动态规划

前言:自从去年noip有了换教室这道概率DP, 这以后各种OJ上的期望概率DP也越来越多,这里简单做个介绍 -------------------------------------------------------------------------------------------------- 文章中涉及例题:hdu 4089,hdu 4405,hdu 4576,poj 2096,poj 3744 [ 1 ] hdu 4405 机器人 要点:基础的概率DP+逆推 题意: 多组输入n,m

2019雅礼集训 D7T1 inverse [概率/期望,DP]

题目描述: 样例: input1: 3 1 1 2 3 output1: 833333340 input2: 5 10 2 4 1 3 5 output2: 62258360 数据范围与约定: 概率/期望的常用套路:将许许多多个元素单独考虑,以达到解决问题的目的. 这里发现不可能整个序列一起考虑,于是枚举任意两个位置,计算出k次翻转之后左边大于右边的概率,再加起来就好了. 于是我们有了一个非常暴力的DP: 令\(dp(i,j,k)?\) 表示k次翻转之后i位置大于j位置的概率.为了方便我们强行令