CodeForces 336 D.Vasily the Bear and Beautiful Strings（组合数学）

Description

定义一个$01$串的权值为，每次把该字符串的最后两位变成一位（如果是两个$0$就变成$1$，否则变成$0$），最后只剩一位即为该字符串的权值，问包含$n$个$0$和$m$个$1$的权为$g$的长度为$n+m$的$01$串个数

Input

三个整数$n,m,g(0\le n,m\le 10^5,n+m\ge 1,0\le g\le 1)$

Output

输出满足条件的字符串数量，结果模$10^9+7$

Sample Input

1 1 0

Sample Output

2

Solution

$n=0$时，如果$m=1$则权为$1$，否则权为$0$

$m=0$时，如果$n$为奇数则权为$0$，否则权为$1$

$n,m\neq 0$时，如果想得到权为$1$的字符串，只有两种情况

$1$.奇数个$0+1+$非空任意数，这样奇数个$0$后面必然多一个$0$变成偶数个$0$，权是$1$

$2$.偶数个$0+1$，$01$变成$0$，进而变成偶数个$0$，权是$1$

分别计数即得到权为$1$的字符串个数，如果问的是权为$0$的字符串，则拿总数$C_{n+m}^n$减去权为 1 <script type="math/tex" id="MathJax-Element-39">1</script>的字符串即可

Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
typedef pair<int,int>P;
const int INF=0x3f3f3f3f,maxn=200005;
#define mod 1000000007
int fact[maxn],inv[maxn];
void init(int n=2e5)
{
    fact[0]=1;
    for(int i=1;i<=n;i++)fact[i]=(ll)i*fact[i-1]%mod;
    inv[1]=1;
    for(int i=2;i<=n;i++)inv[i]=mod-(ll)(mod/i)*inv[mod%i]%mod;
    inv[0]=1;
    for(int i=1;i<=n;i++)inv[i]=(ll)inv[i-1]*inv[i]%mod;
}
int C(int n,int m)
{
    return (ll)fact[n]*inv[m]%mod*inv[n-m]%mod;
}
int main()
{
    init();
    int n,m,g;
    while(~scanf("%d%d%d",&n,&m,&g))
    {
        if(!n)
        {
            if(m==1&&g==1||m>1&&g==0)printf("1\n");
            else printf("0\n");
        }
        else if(!m)
        {
            if(n%2==1&&g==0||n%2==0&&g==1)printf("1\n");
            else printf("0\n");
        }
        else
        {    
            int ans=0;
            for(int i=1;i<=n;i+=2)
                if(n-i+m-1)ans=(ans+C(n-i+m-1,m-1))%mod;
            if(n%2==0&&m==1)ans++;
            if(!g)ans=(C(n+m,m)-ans+mod)%mod;
            printf("%d\n",ans);
        }
    }
    return 0;
}