题目描述
有一束光/那瞬间/是什么痛得刺眼/你的视线是谅解/为什么舍不得熄灭/我逆着光却看见/那是泪光/那力量/我不想再去抵挡/面对希望/逆着光/感觉爱存在的地方/一直就在我身旁
Description
Zyh养着n盆太阳花,它们被排在一列直线上。为了简化问题,太阳花的朝向只有向左和向右这两种。Zyh非常喜欢这些花,于是他在每盆花的上方放置了光源。
太阳花和Zyh幸福地生活着,然而每次要关掉这些光源的时候就会出现一个问题。因为Zyh的动作问题,他每次只能关掉一个光源,在关掉这个光源后,这个光源下方的花就会休眠。然而,对于每一个未休眠的太阳花,如果它的朝向向左,那么每看到一个在它左边的光源关闭,它就会认为自己的人生有些黑暗,于是对Zyh的好感减1。同样地,如果它的朝向向右,那么每看到一个在它右边的光源关闭,也会导致它对Zyh的好感减1。
Zyh不想看到太阳花对他的好感下降太多。于是他想知道关掉所有灯后,太阳花下降的好感总值是多少。
输入
输入包含两行。第一行是一个数正整数n表示太阳花的盆数。然后第二行是n个由空格隔开的0或1。0表示向左,1表示向右。
输出
只有一个数,太阳花下降的好感总值。
样例输入
样例输出
提示
Case two
Input
5
1 0 1 0 1
Output
3
Hint
对于30%的数据 n<10
对于70%的数据 n<=5000
对于100%的数据 n<=1000000
题解
这道题0表示朝左,1表示朝右
首先我们考虑0 1的情况,这样下降的好感度是为0的,而1 0是需要下降1好感度的
再来一个样例1 0 0 0,这样需要下降3好感度,相信大家已经想到了一些
通过这几个样例可以发现
关掉1,下降的好感度是它右边的0的个数
关掉0,下降的好感度是它左边的1的个数
所以不管关0还是1,下降的好感度就是逆序对的个数
所以答案就是逆序对个数,因为数为0或1,所以我们可以预处理
sum[i]表示1~i中1的个数,每次判断朝向为0,ans+=sum[i]即可 注意ans要开long long
#include<bits/stdc++.h>
#define ll long long
#define N 1000005
using namespace std;
int n,x;
ll ans;
int sum[N];
int main(){
scanf("%d",&n);
for (int i=;i<=n;i++){
scanf("%d",&x);
sum[i]=sum[i-]+x;
if (!x) ans+=sum[i];
}
printf("%lld\n",ans);
return ;
}