Codeforces 617B Chocolate (Round #340 (Div. 2) B题)

题意

• 给你一个0、1串，问你有多少中切法，使得每一段恰好有一个1

思路

• 简单的乘法原理，预处理一个数组a，这个数组保存连续的0的个数, 记m为数组大小，数组下标从0开始记
• 比如00110, 数组为2,0,1
• 然后如果数组只有一个元素，说明全是0，ans = 0
• 其它情况，ans = ans * (a[i] + 1) (i = 1到m-2)

实现

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int,int> pii;
#define fi first
#define se second
#define pb push_back
#define mp make_pair
const int inf = 0x3f3f3f3f;

const int maxn = 105;
vector<int> vec;
int a[maxn];

int main(){

    int n,m;
    cin>>n;
    int num = 0;
    for (int i=1;i<=n;i++){
        cin>>a[i];
        if (a[i] == 1){
            vec.pb(num);
            num = 0;
        }
        else{
            num++;
        }
    }
    vec.pb(num);

    if (vec.size() == 1){
        puts("0");
        return 0;
    }
    ll ret = 1;
    for (int i=1;i<vec.size()-1;i++){
        ret *= (ll)(vec[i] + 1);
    }
    cout << ret << endl;
    return 0;
}