[51nod 1667]概率好题

题目描述

甲乙进行比赛。
他们各有k1,k2个集合[Li,Ri]
每次随机从他们拥有的每个集合中都取出一个数
S1=sigma甲取出的数，S2同理
若S1>S2甲胜 若S1=S2平局 否则乙胜
分别求出甲胜、平局、乙胜的概率。
（显然这个概率是有理数，记为p/q，则输出答案为（p/q）%(1e9+7)）（逆元）
注意 多组数据

做一做

把甲每个集合选的数表示为Ri-xi，0<=xi<=Ri-Li
把乙每个集合选的数表示为Li+yi，0<=yi<=Ri-Li
那么现在比如让甲赢，就是
$\sum_{i=1}^{k1}Ri-xi>\sum_{i=1}^{k2}Li+yi$
移项得
$\sum_{i=1}^{k1}xi+\sum_{i=1}^{k2}yi<\sum_{i=1}^{k1}Ri-\sum_{i=1}^{k2}Li$
右边是一个常数，记作m。
因为都是整数， <m <script type="math/tex" id="MathJax-Element-815">
小于等于不太会做，我们引入一个新的变量k，0<=k<= $\infty$
那么
$\sum_{i=1}^{k1}xi+\sum_{i=1}^{k2}yi+k=m-1$
于是我们现在有k1+k2+1个变量，每个变量都有上限限制，让其满足上述等式，求方案数。
我们可以考虑容斥原理，枚举那些变量不符合条件，来得到至少不符合几个条件的方案数。例如对于一个变量xi，不满足条件xi<=upi的话，就是xi>upi，就是xi-upi-1>=0，那我们可以把m减少upi+1，然后xi就变成了没有上限限制。对于不被枚举的那些可能不符合条件的变量，我们也认为它们没有上限。那么就可以组合数求解了。
求平局的话，等式右边是m，然后不需要引入k（或者把k上限设为0）
于是也能求乙赢的情况

#include<cstdio>
#include<algorithm>
#define fo(i,a,b) for(i=a;i<=b;i++)
using namespace std;
typedef long long ll;
const int maxn=20+10,mo=1000000007,inf=1000000000;
int up[maxn];
int i,j,k,l,r,t,n,m,ca,ans,ans2,k1,k2;
int read(){
    int x=0,f=1;
    char ch=getchar();
    while (ch<'0'||ch>'9'){
        if (ch=='-') f=-1;
        ch=getchar();
    }
    while (ch>='0'&&ch<='9'){
        x=x*10+ch-'0';
        ch=getchar();
    }
    return x*f;
}
int quicksortmi(int x,int y){
    if (!y) return 1;
    int t=quicksortmi(x,y/2);
    t=(ll)t*t%mo;
    if (y%2) t=(ll)t*x%mo;
    return t;
}
int C(int n,int m){
    if (n<m) return 0;
    int i,t=1;
    fo(i,n-m+1,n) t=(ll)t*i%mo;
    fo(i,1,m) t=(ll)t*quicksortmi(i,mo-2)%mo;
    return t;
}
void dfs(int x,int y,int m){
    if (m<0) return;
    if (x==k1+k2+2){
        if (y) (ans-=C(m+k1+k2,k1+k2))%=mo;else (ans+=C(m+k1+k2,k1+k2))%=mo;
        return;
    }
    if (x==k1+k2+1){
        dfs(x+1,y,m);
        return;
    }
    dfs(x+1,1-y,m-up[x]-1);
    dfs(x+1,y,m);
}
void dg(int x,int y,int m){
    if (m<0) return;
    if (x==k1+k2+1){
        if (y) (ans-=C(m+k1+k2-1,k1+k2-1))%=mo;else (ans+=C(m+k1+k2-1,k1+k2-1))%=mo;
        return;
    }
    dg(x+1,1-y,m-up[x]-1);
    dg(x+1,y,m);
}
int main(){
    freopen("1667.in","r",stdin);
    scanf("%d",&ca);
    while (ca--){
        m=0;
        k1=read();
        fo(i,1,k1){
            l=read();r=read();
            m+=r;
            up[i]=r-l;
        }
        k2=read();
        fo(i,1,k2){
            l=read();r=read();
            m-=l;
            up[i+k1]=r-l;
        }
        up[k1+k2+1]=inf;
        m--;
        ans=0;
        dfs(1,0,m);
        fo(i,1,k1+k2) ans=(ll)ans*quicksortmi(up[i]+1,mo-2)%mo;
        (ans+=mo)%=mo;
        printf("%d ",ans);
        ans2=ans;
        ans=0;
        m++;
        dg(1,0,m);
        fo(i,1,k1+k2) ans=(ll)ans*quicksortmi(up[i]+1,mo-2)%mo;
        (ans+=mo)%=mo;
        printf("%d %d\n",ans,((1-ans-ans2)%mo+mo)%mo);
    }
}