这题在洛谷居然有紫题,我一激动又跑过来写题解了。
题目大意
给定一个长度为n
的序列,-1
可以被替换成任意正整数,求该序列最少可以被切成多少个等差数列。
样例输入 #1
9
8 6 4 2 1 4 7 10 2
样例输出 #1
3
样例输入 #2
9
-1 6 -1 2 -1 4 7 -1 2
样例输出 #2
3
样例输入 #3
5
-1 -1 -1 -1 -1
样例输出 #3
1
样例输入 #4
7
-1 -1 4 5 1 2 3
样例输出 #4
2
思路
先说一个显而易见的事情:如果在一个数列中枚举到第i
个,它可以和前面的某段数构成等差数列,当然是让它加入前面那一段的等差数列更为划算。形象地说,就是尽可能把等差数列的右端点向右拖。
接下来我们转移视角,如果要确定一个等差数列,就要知道首相和公差。然而-1
十分的麻烦。考虑一种情况,一个数列中找到了两个确定的数,公差就能计算了。假设有a[i]
和a[j]
(j>i),很明显,公差就是(a[j]-a[i])/(j-i)
。(不懂的把小学数学再学一遍) 那么这样就可以避开-1
。
思路逐渐浮出水面,用it
遍历整个数组,用vx
和vy
记录两者的值,x
代表在第一个确定的值前面有几个-1
,y
代表两者之间有几个-1
,模拟一遍样例2。
在上代码之前,先说一说注意点:
往前/后填数的时候,要判断是否为正数 如果在往前填的时候遇到这种情况,只要全部填成一个数(公差为0),例如:2 2 2 2 5 8 11 14,就能在只增加一段的情况下解决问题。
如果找到的两个数算出来的公差不是整数,就不能要,得断成两段 仍旧用笨办法解决,例如:-1 -1 4 -1 5 -1,算出d=0.5
,则填成 4 4 4 4 5 5。
代码
(建议先读代码,忽略continue
这句话)
为了防止可爱的读者们看不懂,解释一下如何做到将一大片都填成同一个数,很简单,只需要continue。遇到问题就continue
。这里是精髓!!!
例:-1 -1 2 -1 6 -1 10 2
当it=5
时,ans=1
,continue
后,it
从第五个开始重新寻找两个数,此时ans=2
。这相当于:2 2 2 2 6 -1 10 2,直接放弃前四个,将它们变成一段,开始找第二段。这刚好应证了我们一开始的贪心思路,将公差为0的这一段尽可能地延长。其他问题的处理也可以自行模拟理解。
#include<bits/stdc++.h>
using namespace std;
int n,it=1,ans;
long long a[200005];
int main(){
scanf("%d",&n);
for(int i=1;i<=n;i++) scanf("%lld",&a[i]);
while(it<=n){
ans++;
int x=0,y=0;
long long vx,vy,d;
while(it<=n&&a[it]==-1) it++,x++;
if(it>n) break;
vx=a[it++];
while(it<=n&&a[it]==-1) it++,y++;
vy=a[it];
if((vy-vx)%(y+1)!=0) continue;//公差不是整数
//y是两个数中间有多少个数,要加1才是两个坐标差了多少
else d=(vy-vx)/(y+1);
if(vx-x*d<=0) continue;//往前填的时候出现了非正数
long long s=vy;
it++;
while(it<=n){
s+=d;
if(s<=0) break;//往后填的时候出现了非正数
if(a[it]==-1) a[it]=s;
else if(a[it]!=s) break;//遇到不属于同一段的数
it++;
}
}
printf("%d\n",ans);
return 0;
}