传送门: https://ac.nowcoder.com/acm/contest/53366/D
题意:找到一个区间段只含有一个1其余都是0
思路:记录下当前的1之前有多少个0,和当前1后有多少个0
代码
#include <iostream>
#include <vector>
#include <cstdio>
#include <cmath>
#include <algorithm>
#include <list>
#include <map>
#include <vector>
#include <queue>
#include <stack>
#define IOS() ios::sync_with_stdio(false),cin.tie(0),cout.tie(0)
#define int long long
#define fir(v) (v).first
#define se(v) (v).second
#define pb(x) push_back(x)
#define max(a, b) ((a) > (b) ? (a) : (b))
#define min(a, b) ((a) < (b) ? (a) : (b))
#define pii pair<int,int>
#define all(a) (a).begin(),(a).end()
#define mp(a,b) make_pair(a,b)
using namespace std;
const int N=1000000;
int a[N],b[N];
bool cmp(int x,int y)
{
return x>y;
}
signed main()
{
IOS();
int n;
cin>>n;
int temp;
int sum=0;
int s=0,l=0;
//s记录当前1的位置与上一个1的位置之间的距离,
// 也就是左侧所有可能的孤独区间数
for(int i=1;i<=n;i++){
cin>>temp;
if(temp){//则当前位置的s其左侧孤独区间总数
s=i-l;//孤独区间数
l=i;//每一次记录1的位置
}
sum+=s;//每次sum
}
cout<<sum<<endl;
}