bzoj 5368: [Pkusc2018]真实排名 线段树+组合数

题意

小C是某知名比赛的组织者，该比赛一共有n名选手参加，每个选手的成绩是一个非负整数，定义一个选手的排名是
：成绩不小于他的选手的数量（包括他自己）。例如如果333位选手的成绩分别是[1,2,2]，那么他们的排名分别是
[3,2,2]。拥有上帝视角的你知道所有选手的实力，所以在考试前就精准地估计了每个人的成绩，设你估计的第iii
个选手的成绩为Ai,且由于你是上帝视角，所以如果不发生任何意外的话，你估计的成绩就是选手的最终成绩。但
是在比赛当天发生了不可抗的事故（例如遭受到了外星人的攻击），导致有一些选手的成绩变成了最终成绩的两倍
，即便是有上帝视角的你也不知道具体是哪些选手的成绩翻倍了，唯一知道的信息是这样的选手恰好有k个。现在
你需要计算，经过了不可抗事故后，对于第i位选手，有多少种情况满足他的排名没有改变。由于答案可能过大，
所以你只需要输出答案对998244353取模的值即可。

分析

讨论一波就好了啊，有点细节
就是每一次取不取这个数
最后我的做法要这个数是0的时候判断一下
其实也可以不用线段树。。

代码

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll N = 3000010;
const ll inf = 1e9;
const ll Mod = 998244353;
inline ll read()
{
  ll p=0; ll f=1; char ch=getchar();
  while(ch<'0' || ch>'9'){if(ch=='-') f=-1; ch=getchar();}
  while(ch>='0' && ch<='9'){p=p*10+ch-'0'; ch=getchar();}
  return p*f;
}
ll rt,tot = 0,lc[N],rc[N],c[N],a[N];
void link(ll &u,ll L,ll R,ll k)
{
  if(!u) u = ++tot;
  if(L==R){c[u] ++; return ;}
  ll mid=(L+R)>>1;
  if(k<=mid) link(lc[u],L,mid,k);
  else link(rc[u],mid+1,R,k);
  c[u] = c[lc[u]] + c[rc[u]];
}
ll qry(ll u,ll L,ll R,ll l,ll r)
{
  if(l>r) return 0;
  if(L==l && R==r) return c[u];
  ll mid=(L+R)>>1;
  if(r<=mid) return qry(lc[u],L,mid,l,r);
  else if(l>mid) return qry(rc[u],mid+1,R,l,r);
  else return qry(lc[u],L,mid,l,mid) + qry(rc[u],mid+1,R,mid+1,r);
}
ll inv[N],fac[N];
ll C(ll x,ll y){if(x<y || x<0 || y<0) return 0; return fac[x] * inv[y] % Mod * inv[x-y] % Mod;}
int main()
{
  ll n = read(); ll k = read();
  rt=tot=0; for(ll i=1;i<=n;i++) link(rt,0,inf,a[i] = read());
  fac[0] = 1; for(ll i=1;i<=n;i++) fac[i] = i;
  inv[0] = inv[1] = 1; for(ll i=2;i<=n;i++) inv[i] = (Mod - Mod/i) * inv[Mod % i] % Mod;
  for(ll i=1;i<=n;i++) fac[i] = fac[i-1] * fac[i] % Mod,inv[i] = inv[i-1] * inv[i] % Mod;
  for(ll i=1;i<=n;i++)
  {
    if(!a[i]){printf("%lld\n",C(n,k)); continue;}
    ll s1 = qry(rt,0,inf,(a[i]+1)/2,a[i]-1)+1; s1 = n - s1;
    ll s2 = qry(rt,0,inf,a[i],min(inf,a[i]*2-1));
    printf("%lld\n",(C(s1,k) + ((s2<=k) ? C(n-s2,k-s2) : 0))%Mod);
  }
  return 0;
}