drink

drink

题目背景

题目描述

你面前有 $n$ 个瓶子，从左到右依次编号为 $1\sim n$，瓶子可能是空的也可能装着水。

你可以选择一对 $l,r(l\le r)$，然后把 $l\sim r$ 号瓶子中的所有水全部喝完。你想一口气把桌上剩的水全喝完，请问你最少要拿起几个瓶子？

你有可能一个瓶子都不需要拿起来。

输入格式

第一行一个整数 $n$。

第二行 $n$ 个整数，第 $i$ 个整数为 $1$ 表示第 $i$ 个瓶子装着水，为 $0$ 表示第 $i$ 个瓶子是空的。

输出格式

一行一个整数 $k$，表示最少拿起的瓶子数。

样例 #1

样例输入 #1

5
0 0 0 1 0

样例输出 #1

1

样例 #2

样例输入 #2

6
0 0 1 1 0 1

样例输出 #2

4

提示说明

【样例说明】

样例 $1$ 中，拿起 $4$ 号瓶子即可。总共拿起 $1$ 个瓶子。

样例 $2$ 中，拿起 $3\sim 6$ 号瓶子可以将所有的水喝完。总共拿起 $4$ 个瓶子。

【数据范围】

对于 $30\%$ 的数据，保证 $n\le 100$。

对于 $60\%$ 的数据，保证 $n\le 2000$。

对于 $100\%$ 的数据，保证 $1\le n\le 10^5$，$0\le a_i\le 1$。

代码内容

// #include <iostream>
// #include <algorithm>
// #include <cstring>
// #include <stack>//栈
// #include <deque>//队列
// #include <queue>//堆/优先队列
// #include <map>//映射
// #include <unordered_map>//哈希表
// #include <vector>//容器，存数组的数，表数组的长度
#include <bits/stdc++.h>

using namespace std;

typedef long long ll;
const ll N=1e5+10;
ll a[N];

int main()
{
    ll n,l,r;
    cin>>n;

    bool x=false;
    for(ll i=0;i<n;++i)
    {
        cin>>a[i];
        //若出现1，则用x=true标记
        if(a[i]==1) x=true;
    }

    //特判所有瓶子都没水的情况
    if(x==false)
    {
        cout<<0<<endl;
        return 0;
    }
    
    for(ll i=0;i<n;i++)
    {
        if(a[i]==1)
        {
            l=i;
            break;
        }
    }
    
    for(ll i=n-1;i>=0;i--)
    {
        if(a[i]==1)
        {
            r=i;
            break;
        }
    }
    
    cout<<r-l+1<<endl;
    return 0;
}