「PKUSC2018」真实排名 (简单组合数)

最新推荐文章于 2023-05-08 15:40:05 发布

偶耶XJX

最新推荐文章于 2023-05-08 15:40:05 发布

阅读量212

点赞数

分类专栏：信息竞赛解题

本文链接：https://blog.csdn.net/weixin_43960287/article/details/108499164

版权

信息竞赛解题专栏收录该内容

194 篇文章 6 订阅

订阅专栏

题面：「PKUSC2018」真实排名

题解

分类讨论，对于第 i 个人：

若 $a_{i}$ 不翻倍，则对于满足 $a_{j}<a_{i},a_{j}*2\geq a_{i}$ 的任意 j ，第 j 个人必须不翻倍，否则第 i 个人的排名一定变，因为其他人无论翻不翻倍，对 i 的排名没有影响。设这样满足条件的 j 的数量为 m ，那么方案数就为组合数 $C_{n-m-1}^{k}$ ；

若 $a_{i}$ 翻倍，则对于满足 $a_{i}\leq a_{j}<a_{i}*2,j\neq i$ 的任意 j ，第 j 个人必须翻倍，原因同理。设这样满足条件的 j 的数量为 p ，那么方案数就为组合数 $C_{n-p-1}^{k-p-1}$ ；

所以第 i 个人的总方案数为 $C_{n-m_{i}-1}^{k}+C_{n-p_{i}-1}^{k-p_{i}-1}$ 。

提示：代码中慎用快速乘！因为我把考试源代码删了快速乘就A了(QAQ)

代码

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<vector>
#include<cmath>
#define ll long long
#define MAXN 100005
#define mod 998244353ll
using namespace std;
inline ll read(){
	ll x=0,f=1;char s=getchar();
	while(s<'0'||s>'9'){if(s=='-')f*=-1;s=getchar();}
	while(s>='0'&&s<='9'){x=(x<<1)+(x<<3)+s-'0';s=getchar();}
	return x*f;
}
int n,k;
struct itn{
	ll x;int y;
	itn(){}
	itn(ll X,int Y){x=X,y=Y;}
}a[MAXN];
ll jc[MAXN],ans[MAXN];
inline bool cmp(itn a,itn b){return a.x<b.x;}
inline ll ksm(ll a,ll b,ll mo){
	ll res=1;
	for(;b;b>>=1){
		if(b&1)res=res*a%mod;
		a=a*a%mod;
	}
	return res;
}
inline ll C(int n,int m){
	if(m>n||m<0)return 0;
	return jc[n]*ksm(jc[m],mod-2,mod)%mod*ksm(jc[n-m],mod-2,mod)%mod;
}
int main()
{
//	freopen("rank.in","r",stdin);
//	freopen("rank.out","w",stdout);
	n=read(),k=read(),jc[0]=1;
	for(int i=1;i<MAXN-4;i++)jc[i]=jc[i-1]*i%mod;
	for(int i=1;i<=n;i++)a[i]=itn(read(),i);
	sort(a+1,a+1+n,cmp);
	for(int i=1;i<=n;i++){
		int id=i,t;
		for(int j=17,o=id-(1<<j);j>=0;j--,o=id-(1<<j))
			if(o>0&&a[o].x==a[id].x)id=o;
		t=id;
		for(int j=17,o=t-(1<<j);j>=0;j--,o=t-(1<<j))
			if(o>0&&(a[o].x<<1)>=a[id].x)t=o;
		ans[a[i].y]=C(n-id+t-1,k)%mod,t=id;
		for(int j=17,o=t+(1<<j);j>=0;j--,o=t+(1<<j))
			if(o<=n&&a[o].x<(a[id].x<<1))t=o;
		(ans[a[i].y]+=C(n-t+id-1,k-t+id-1)%mod)%=mod;
	}
	for(int i=1;i<=n;i++)printf("%lld\n",ans[i]);
	return 0;
}