题目链接:http://codeforces.com/problemset/problem/416/D
昨天比赛的题,当时感觉比较麻烦,后来看完别人的博客,用函数实现思路比较清晰,另外公差可以为0,尴尬。。。
题目大意:给你一个由正整数和-1组成的序列,-1可以表示任意正整数,问最少可以划分成多少个等差数列,如果一个数既可以在前面的数列中,又可以在后面的数列中,则让它在后面的数列中最优,因为这样首先不会增加等差数列的个数,又可以尽可能减少它对后面数列的影响,故贪心最大长度。。。
代码如下:解释在代码中
#include<stdio.h>
#define N 200005
__int64 s[N],n;
//查找出数列中两个不为零的位置,如果找不到返回数列结尾位置,即n+1
__int64 find(__int64 x)
{
__int64 i;
for(i=x;i<=n;i++)
if(s[i]!=-1)
return i;
return n+1;
}
__int64 judge(__int64 x)
{
__int64 mark1=find(x);
__int64 mark2=find(mark1+1);
//如果数列中只有一个整数,则可构成公差为0的等差数列,故此时数列为一部分,直接返回结尾
if(mark1==n+1||mark2==n+1)
return n+1;
//如果两个整数之间的数不能与这两个整数构成等差数列,则让mark2之前的够成一个数列
if((s[mark2]-s[mark1])%(mark2-mark1)!=0)
return mark2;
//如果可以构成,求出公差d
__int64 d=(s[mark2]-s[mark1])/(mark2-mark1);
//判断mark1前面的能否和mark1构成等差数列,不能的话,就让mark2之前的够成一个数列
if(s[mark1]-d*(mark1-x)<1)
return mark2;
//如果都可以合并为一个数列,则暴力找到这个数列的最长位置
__int64 sum=s[mark2];
for(__int64 i=mark2+1;i<=n;i++)
{
sum+=d;
if(sum<1||s[i]!=-1&&s[i]!=sum)
return i;
}
return n+1;
}
__int64 solve()
{
__int64 ans=0,x=1;
//每次更新序列的结尾位置,知道序列遍历玩
while(x<=n)
{
ans++;
x=judge(x);
}
return ans;
}
int main()
{
__int64 i,j;
while(~scanf("%I64d",&n))
{
for(i=1;i<=n;i++)
scanf("%I64d",&s[i]);
printf("%I64d\n",solve());
}
}