CF1983E I Love Balls

本文同步发布于我的网站，欢迎来看看！

CF1983E I Love Balls

Problem - E - Codeforces

爱丽丝和鲍勃玩摸球游戏。有 $n$ 个球，其中 $k$ 个是特殊球。每个球都有其价值。

他们轮流且不放回地摸球，每回合随机摸一个球并获得该球的价值。特别地，如果摸到了特殊球（且至少还有一个球）则这名玩家继续摸球。如果摸到的是普通球，则换人摸球。这样轮流摸球直到没有剩余球，游戏结束。Alice先手。

求游戏结束时双方的期望得分，对取模 $10^9+7$ 。

$t\le 2\times 10^5,1\le k\le n\le 4\times 10^5,1\le v_i\le 10^7,\sum n\le 5\times 10^5$

Solution

看别人题解的时候看见一个很常见的结论，先记录一下

取球的过程可以看作是n个球提前排列好，然后依次去取。

所以取到每个普通球的概率都是相同的（特殊球同理）。

我们尝试模拟一下两个人做摸球游戏，发现如果摸到了特殊球就会一直摸下去，直到遇到了普通球或者摸完了为止。摸到了普通球就换人，摸完了则游戏结束。

先考虑普通球的贡献。由于某个玩家摸到了特殊球就会继续摸，直到他摸到了普通球（或者没球了）换人。所以这 $n-k$ 个普通球一定是交替分配给两人的。Alice将会得到奇数位上的所有普通球，对于某个普通球而言被分配到奇数位上的概率为 $\frac{\lceil \frac{n-k}{2}\rceil }{n-k}$；Bob将会得到偶数位上的所有普通球，对应概率为 $\frac{\lfloor \frac{n-k}{2}\rfloor }{n-k}$​。

接着考虑特殊球。$n-k$ 个普通球产生了$n-k+1$ 个回合区间，每个特殊球都会等概率的出现在这 $n-k+1$ 个区间中。其中奇数回合内的所有球由Alice得到，概率为 $\frac{\lceil \frac{n-k+1}{2}\rceil }{n-k+1}$；偶数回合内的所有球由Bob得到，概率为 $\frac{\lfloor \frac{n-k+1}{2}\rfloor }{n-k+1}$。

最终期望为每个球的价值乘以摸到的概率。

Code

#define N 1000010
#define p 1000000007ll

LL n,k;
LL a[N];

LL qpow(LL a,LL b)
{
	LL ans=1;
	while(b)
	{
		if(b&1) ans=ans*a%p;
		a=a*a%p;
		b>>=1;
	}
	return ans%p;
}

LL inv(LL x)
{
	return qpow(x,p-2);
}

int main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);
	cout.precision(10);
	int t=1;
	cin>>t;
	while(t--)
	{
		cin>>n>>k;
		for(int i=1;i<=n;i++)
		{
			cin>>a[i];
		}
		LL ans1=0,ans2=0,inv1=inv(n-k),inv2=inv(n-k+1);
		for(int i=k+1;i<=n;i++)
		{
			ans1=(ans1+(n-k+1)/2*a[i]%p*inv1)%p;
			ans2=(ans2+(n-k)/2*a[i]%p*inv1)%p;
		}
		for(int i=1;i<=k;i++)
		{
			ans1=(ans1+(n-k+2)/2*a[i]%p*inv2)%p;
			ans2=(ans2+(n-k+1)/2*a[i]%p*inv2)%p;
		}
		cout<<ans1<<" "<<ans2<<endl;
	}
	return 0;
}