D. Rescue Nibel! 解析(思維、組合、離散化、差分)

Codeforce 1420 D. Rescue Nibel! 解析(思維、組合、離散化、差分)

今天我們來看看CF1420D
題目連結

題目
給你(n)個區間，求有幾種方法使得(k)個區間的交集非空。

前言

組合不會算，也想不到離散化

@copyright petjelinux 版權所有
觀看更多正版原始文章請至petjelinux的blog

想法

首先需要找個依據來枚舉開始計算，而我們可以觀察到:對於任何一個(k)個區間的交集，這個交集的左界一定是某個區間的左界，也就是說我們可以枚舉交集所有可能的左界，把答案加總即可。
而假設要計算交集從(i)開始的方法數，我們必須知道究竟有多少個區間有包含這個左界，但是(l_i,r_ile10^9)實在太大了，即使我們做差分也時間不夠，因此我們需要離散化整個座標軸。

差分即是:(cnt[左界]++,cnt[右界+1]--)，如此一來只要把整個(cnt)數列做前綴和，其結果就是每個點被覆蓋的次數。

離散化:我們先把所有(l_i,r_i)丟進一個(vector)裡，並且只留下相異元素、排序，如此一來某原始座標(x)的離散化後的座標即是(lower\_bound(vector_{start},vector_{end},x))。

我們還需要紀錄:對於每一個座標，有多少左界從這開始。
如此一來，我們只要遍歷所有座標點，答案加上:(覆蓋的區間中選(k)個的方法數(-)沒選到從當前座標開始的區間的方法數)，就可以算出答案。

而還有一個難點即是計算組合數。我們可以先愈處理所有(x!)的數值和模反元素(計算模反元素可以用Fermat's Little Theorem:(a^{p-1}equiv1mod p)，因為(998244353)是質數，所以(a^{p-2}equiv a^{-1}mod p))，接著就用一般的公式計算即可。

程式碼:

const int _n=3e5+10;
int t,n,k,cnt[_n<<1],num[_n<<1];
PII la[_n];
VI v;
int fac[_n],inv[_n];
void exgcd(int a,int b,int& d,int& x,int& y){
  if(!b)x=1,y=0,d=a;
  else exgcd(b,a%b,d,y,x),y=(1ll*y-1ll*x*(a/b)%mod+mod)%mod;
}
int C(int m,int n){
  if(m<n)return 0;
  if(m<mod and n<mod)return 1ll*fac[m]*inv[n]%mod*inv[m-n]%mod;
  return 1ll*C(m/mod,n/mod)*C(m%mod,n%mod)%mod;
}
void genInv(){
  fac[0]=1;rep(i,1,n+1)fac[i]=1ll*fac[i-1]*i%mod;
  //int tmp1,tmp2;rep(i,0,n+1)exgcd(fac[i],mod,tmp1,inv[i],tmp2);
  inv[n]=powmod(fac[n],mod-2);per(i,0,n)inv[i]=1ll*inv[i+1]*(i+1)%mod;
}
main(void) {ios_base::sync_with_stdio(0);cin.tie(0);cout.tie(0);
  cin>>n>>k;rep(i,0,n){cin>>la[i].fi>>la[i].se;v.pb(la[i].fi),v.pb(la[i].se);}
  sort(all(v));int nn=unique(all(v))-v.begin(); genInv(); ll ans=0;
  rep(i,0,n){
    la[i].fi=lower_bound(v.begin(),v.begin()+nn,la[i].fi)-v.begin();
    la[i].se=lower_bound(v.begin(),v.begin()+nn,la[i].se)-v.begin();
  }rep(i,0,n)cnt[la[i].fi]++,cnt[la[i].se+1]--,num[la[i].fi]++;
  rep(i,1,nn)cnt[i]=cnt[i-1]+cnt[i];
  rep(i,0,nn)ans=(ans+C(cnt[i],k)-C(cnt[i]-num[i],k)+mod)%mod;
  cout<<ans<<'
';
  return 0;
}

標頭、模板請點Submission看((exgcd)用不到，且(C)函數我寫的是Lucas定理法)
Submission