题目描述
题目描述
给定一个长度是 n n n的数列 A A A,我们称一个数列是完美的,当且仅当对于其任意连续子序列的和都是正的。现在你有一个操作可以改变数列,选择一个区间[X,Y]满足 A x + A x + 1 + ⋯ + A Y , 1 < X ≤ Y < n A_x +A_{x+1} + \cdots + A_Y, 1<X \leq Y<n Ax+Ax+1+⋯+AY,1<X≤Y<n,令 S = A x + A x + 1 + … + A Y S=A_x +A_{x+1} +…+ A_Y S=Ax+Ax+1+…+AY,对于 A x − 1 A_{x-1} Ax−1和 A Y + 1 A_{Y+1} AY+1分别加上 S S S, A x A_x Ax和 A Y A_Y AY分别减去 S S S(如果 X = Y X=Y X=Y就减两次)。问最少几次这样的操作使得最终数列是完美的。
输入格式
第一行一个数 n n n,以下 n n n个数。
【数据规模】
对于 20 20 20%的数据,满足 1 ≤ N ≤ 5 1 \leq N \leq 5 1≤N≤5;
对于 100 100 100%的数据,满足 1 ≤ N ≤ 1 0 5 ; 1 ≤ ∣ A [ i ] ∣ ≤ 2 31 − 1 1≤N≤10^5; 1≤|A[i] |≤2^{31}-1 1≤N≤105;1≤∣A[i]∣≤231−1。
输出格式
一个数表示最少的操作次数,如果无解输出 − 1 -1 −1。
输入输出样例
输入
5
13
-3
-4
-5
62
输出
2
说明/提示
【样例解释】
首先选择区间 [ 2 , 4 ] [2,4] [2,4],之后数列变成 1 , 9 , − 4 , 7 , 50 1,9,-4,7,50 1,9,−4,7,50,然后选择 [ 3 , 3 ] [3,3] [3,3],数列变成 1 , 5 , 4 , 3 , 50 1,5,4,3,50 1,5,4,3,50
思路
一开始的时候,我看到这道题并没有什么思路,或者说是就是在空想,各种想思路,各种被自己hack,之后,自己就自闭了……
之后,就请教了MILLOPE 孟神。
首先,我们要思考的是:对于其任意连续子序列的和
你会想到怎么求解??
线段树??DP??
当然,都不是,神仙告诉我说,我们要想到的应该是前缀和,前缀和是支持O(1)求解任何连续区间的和的。
然后,我们开始把思路向前缀和靠拢,我们可以把样例的前缀和算一下:
原序列:13 -3 -4 -5 62
前缀和:13 10 6 1 63
修改一次:
原序列:1 9 -4 7 50
前缀和:1 10 6 13 63
有没有发现什么呢??
修改两次:
原序列:1 5 4 3 50
前缀和:1 6 10 13 63
我们可以发现的是:在多次修改的过程中,我们的前缀和是并没有发生改变的,而是不断的在改变自己的位置,直到把自己的位置变得有序为止。
那,我们考虑一下,前缀和是怎么在调换位置呢??
令S为连续区间的和:
S
=
∑
i
=
x
y
a
[
i
]
S=\sum_{i=x}^ya[i]
S=∑i=xya[i];
令
s
u
m
i
sum_i
sumi为前缀和数组:
s
u
m
[
i
]
=
∑
k
=
1
i
sum[i] =\sum_{k=1}^i
sum[i]=∑k=1i;
从上面的描述中,我们可以发现每次操作只会影响前缀和。
即:
s
u
m
[
x
−
1
]
+
=
S
sum[x-1]+=S
sum[x−1]+=S,
s
u
m
[
x
]
+
=
0
sum[x]+=0
sum[x]+=0;
s
u
m
[
y
]
−
=
S
sum[y]-=S
sum[y]−=S,
s
u
m
[
y
+
1
]
+
=
0
sum[y+1]+=0
sum[y+1]+=0。(前缀和数组是会有累计价值的)
实际上最终影响的只有
s
u
m
[
x
−
1
]
sum[x-1]
sum[x−1],
s
u
m
[
y
]
sum[y]
sum[y]而因为原本的
s
u
m
[
x
−
1
]
+
S
=
s
u
m
[
y
]
sum[x-1]+S=sum[y]
sum[x−1]+S=sum[y]所以其实就相当于交换
s
u
m
[
x
−
1
]
,
s
u
m
[
y
]
sum[x-1],sum[y]
sum[x−1],sum[y]。
所以,前缀和改变位置的方法就是两个前缀和进行交换。
那么题目转化为给定一个序列每次操作可交换序列中任意两个数,求使得序列单调递增的最小交换次数。
我们可以 O ( n ) O(n) O(n) 扫一遍即可,最终复杂度为 O ( n l o g n ) O(nlogn) O(nlogn)
code
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int nn=1e5+3;
ll ans=0;
int n, a[nn], sum[nn];
inline int read()
{
int x=0,f=1; char ch=getchar();
while(ch>'9'||ch<'0'){if(ch=='-') f=-1;ch=getchar();}
while(ch>='0'&&ch<='9'){x=x*10+ch-'0'; ch=getchar();}
return x*f;
}
int main()
{
n=read();
for(int i=1;i<=n;++i)
{
a[i]=sum[i]=sum[i-1]+read();
if(a[i]<0){puts("-1"); return 0;}
}
sort(a+1,a+1+n);
int m=unique(a+1,a+1+n)-(a+1);
for(int i=1;i<=n;++i)
sum[i]=lower_bound(a+1,a+1+m,sum[i])-a;/*离散化*/
for(int i=1;i<=n;++i)
{
while(sum[i]!=i)
{
int x=sum[i];
int temp=sum[x];
sum[x]=sum[i], sum[i]=temp;
ans++;
}
}
printf("%lld\n", ans);
return 0;
}
完~