D - Rescue Nibel!(差分+組合數學+思維)詳解

2020-09-27 09:00:54

https://codeforces.com/contest/1420/problem/D


題意:問有多少種方案可以滿足一段時間裡面有k盞燈亮著。

思路:首先把區間的問題轉化到差分上去。考慮用差分代替區間。

然後看樣例

5 2
1 3
2 4
3 5
4 6
5 7

發現很符合區間覆蓋的貪心樣子。按照左端點排序。然後用一個cnt變數表示列舉到當前l時候前面還有多少能選的,cnt+=差分,如果當前列舉到的點差分值為1,說明在前面還亮著的取C(cnt,k-1)個就是答案。列舉就好了。

程式碼在本地出現異常..交上去ac.如果哪位懂的話感謝指出。

「terminate called after throwing an instance of 'std::bad_alloc'」

#include<iostream>
#include<vector>
#include<queue>
#include<cstring>
#include<cmath>
#include<map>
#include<set>
#include<cstdio>
#include<algorithm>
#define debug(a) cout<<#a<<"="<<a<<endl;
using namespace std;
const int maxn=3e5+100;
typedef long long ll;
using namespace std;
const int inf=0x3f3f3f3f;
const int mod=998244353;
ll fac[maxn];
ll p=mod;
struct node{
	ll x,c;
	bool operator<(const struct node &T)const{
		if(x==T.x) return c<T.c;
		return x<T.x;
	} 
};
vector<node>q;
//bool cmp(node A,node B)
//{
//	if(A.x==B.x) return A.c<B.c;
//	return A.x<B.x;
//}
ll quick(ll a,ll b){
    ll ans=1;
    while(b){
        if(b&1)ans=ans*a%p;
        a=a*a%p;
        b/=2;
    }
    return ans%p;
}
ll ccc(ll n,ll m){//求組合數
	if(m>n) return 0;
	return (fac[n] * quick(fac[m], p - 2) % p * quick(fac[n - m], p - 2) % p)%p ;
}
int main()
{
    cin.tie(0);std::ios::sync_with_stdio(false);
	//ccc(4,3);
	fac[0] = 1;
    for (ll i = 1; i <= maxn; i++){
            fac[i] = fac[i - 1] * i % p;
    }
    ll n,k;cin>>n>>k;
	for(ll i=1;i<=n;i++)
    {
    	ll l,r;cin>>l>>r;
    	q.push_back({l,1});
    	q.push_back({r+1,-1});
	}
	sort(q.begin(),q.end());
	ll cnt=0;ll res=0;//cnt表示當前列舉到的l前面還有多少個亮著的 
	for(auto i:q)
	{
		if(i.c==1) res=(res%mod+ccc(cnt,k-1)%mod)%mod;
		cnt+=i.c;
	}
	cout<<res<<endl;
return 0;
}