CodeForces 617 B. Chocolate（组合数学）

Description

给出一个只由01组成的序列，要求将这个序列分成若干段，每段仅含一个1，问有多少种划分方法

Input

第一行为一整数n表示序列长度，之后n个整数表示该序列(1<=n<=100)

Output

输出划分方案数

Sample Input

5

1 0 1 0 1

Sample Output

4

Solution

简单计数，记录每个1的位置pos[i],假设有res个1，那么ans=(pos[2]-pos[1])(pos[3]-pos[2])…*(pos[res]-pos[res-1])

Code

#include<cstdio>
#include<iostream>
using namespace std;
#define maxn 111
int n,a[maxn],res,pos[maxn];
long long ans;
int main()
{
while(~scanf("%d",&n))
{
res=0;ans=1ll;
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i]);
if(a[i]==1)pos[++res]=i;
}
if(res==0)ans=0;
for(int i=1;i<res;i++)
ans*=(pos[i+1]-pos[i]);
printf("%I64d\n",ans);
}
return 0;
}