BZOJ_5368_[Pkusc2018]真实排名_组合数
Description
小C是某知名比赛的组织者，该比赛一共有n名选手参加，每个选手的成绩是一个非负整数，定义一个选手的排名是：成绩不小于他的选手的数量（包括他自己）。例如如果333位选手的成绩分别是[1,2,2]，那么他们的排名分别是[3,2,2]。拥有上帝视角的你知道所有选手的实力，所以在考试前就精准地估计了每个人的成绩，设你估计的第iii个选手的成绩为Ai,且由于你是上帝视角，所以如果不发生任何意外的话，你估计的成绩就是选手的最终成绩。但是在比赛当天发生了不可抗的事故（例如遭受到了外星人的攻击），导致有一些选手的成绩变成了最终成绩的两倍，即便是有上帝视角的你也不知道具体是哪些选手的成绩翻倍了，唯一知道的信息是这样的选手恰好有k个。现在你需要计算，经过了不可抗事故后，对于第i位选手，有多少种情况满足他的排名没有改变。由于答案可能过大，所以你只需要输出答案对998244353取模的值即可。
Input
第一行两个正整数n,k
第二行n个非负整数A1..An
1≤k<n≤10^5 ，0≤Ai≤10^9
Output
输出n行，第i行一个非负整数ansi，表示经过不可抗事故后，第i位选手的排名没有发生改变的情况数。
Sample Input
3 2
1 2 3
1 2 3
Sample Output
3
1
2
样例解释
一共有3种情况：(1,2)翻倍，(1,3)翻倍，(2,3)翻倍。
对于第一个选手来说，他的成绩就算翻倍，其他人都不低于他，所以任意情况下他的排名都不会改变。
对于第二个选手来说，如果是(1,2)翻倍，成绩变成(2,4,3)，他的排名变成了第一；
如果是(1,3)翻倍，则成绩变成(2,2,6)，他的排名变成了第三；如果是(2,3)翻倍，则成绩变成(1,4,6)，他的排名还是第二。
所以只有一种情况。
对于第三个选手来说，如果是(1,2)翻倍，他的排名会变成第二，其他情况下都还是第一。
GGGGGGGGGGGGGGGGGGGGGGGGGGGGGGG
调了一晚上！！！最后发现组合数求错了！！没判m<0的情况！！那应该RE啊怎么一直WA啊...
求出每个人的排名rank然后分这个人是否乘2进行讨论，设这个人的分数为w。
每次求出 原来小于w，乘2后也小于w的个数p1.原来大于等于w，乘2后也大于等于w的个数p3，其他的人数p2。设第三种人选了q2个人。
有p3+q2=rank-1
那么答案为C(p2)(q2)*C(p1+p3)(K-q2)或C(p2)(q2)*C(p1+p3)(K-q2-1)。
注意当a[i]=0时直接令答案为C(n)(K)即可，省的在求p3的时候特判。
代码：
#include <cstdio>
#include <string.h>
#include <algorithm>
using namespace std;
typedef long long ll;
#define mod 998244353
#define N 100050
int n,K;
ll fac[N<<1],inv[N<<1],ans[N];
int t1[N],t2[N],p1,p2,p3,Rank[N],a[N];
ll qp(ll x,ll y) {
ll re=1; for(;y;y>>=1ll,x=x*x%mod) if(y&1ll) re=re*x%mod; return re;
}
void init() {
int i;
for(fac[0]=1,i=1;i<=n+n;i++) fac[i]=fac[i-1]*i%mod;
inv[n+n]=qp(fac[n+n],mod-2);
for(i=n+n-1;i>=0;i--) inv[i]=inv[i+1]*(i+1)%mod;
}
ll C(ll n,ll m) {
if(m<0||n<m) return 0;
return fac[n]*inv[m]%mod*inv[n-m]%mod;
}
int main() {
scanf("%d%d",&n,&K);
init();
int i;
for(i=1;i<=n;i++) {
scanf("%d",&a[i]); t1[i]=a[i]; t2[i]=a[i]<<1;
}
sort(t1+1,t1+n+1);
sort(t2+1,t2+n+1);
for(i=1;i<=n;i++) {
Rank[i]=n-(lower_bound(t1+1,t1+n+1,a[i])-t1)+1;
}
for(i=1;i<=n;i++) {
if(a[i]==0) {ans[i]=C(n,K); continue;}
p1=lower_bound(t2+1,t2+n+1,a[i])-t2-1;
p3=n-(lower_bound(t1+1,t1+n+1,a[i])-t1);
p2=n-1-p1-p3;
int q2=Rank[i]-1-p3;
ans[i]=C(p2,q2)*C(n-1-p2,K-q2)%mod;
}
for(i=1;i<=n;i++) {
if(a[i]==0) continue;
p1=lower_bound(t2+1,t2+n+1,a[i]<<1)-t2-1;
p3=n-(lower_bound(t1+1,t1+n+1,a[i]<<1)-t1)+1;
p2=n-1-p1-p3;
int q2=Rank[i]-1-p3;
ans[i]=(ans[i]+C(p2,q2)*C(n-1-p2,K-q2-1)%mod)%mod;
}
for(i=1;i<=n;i++) printf("%lld
",ans[i]);
}