CodeForces 416D （贪心）

Problem Population Size

题目大意

　　给一个长度为n的序列，由 -1 和正整数组成，-1表示任意的正整数。

　　将序列分成若干段，使得任意段都是等差数列，求最少段数。

解题分析

　　可以发现对于某一段序列，越长越好。贪心加点，保证每段都是最长就可以了。

　　Tips：一段相同的数也可以算是等差数列。

参考程序

#include <map>
#include <set>
#include <stack>
#include <queue>
#include <cmath>
#include <ctime>
#include <string>
#include <vector>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cassert>
#include <iostream>
#include <algorithm>
#pragma comment(linker,"/STACK:102400000,102400000")
using namespace std;

#define N 200008             
#define M 50008    
#define LL long long
#define lson l,m,rt<<1
#define rson m,r+1,rt<<1|1 
#define clr(x,v) memset(x,v,sizeof(x));
const int mo  = 1000000007;
const int inf = 0x3f3f3f3f;
const int INF = 2000000000;
/**************************************************************************/ 
int n;
int a[N],r[N];

int main(){
    scanf("%d",&n);
    for (int i=1;i<=n;i++) scanf("%d",&a[i]);
    int x=n+1;
    for (int i=n;i>=1;i--){
        r[i]=x;
        if (~a[i]) x=i;
    }

    int i=1,j,k,d,num1,num2,ans=0;
    
    while (i<=n){
        //printf("%d
",i );
        if (a[i]==-1){
            j=r[i]; k=r[j];
            if (j==n+1) { ans++; break; } else num1=j-i; 
            num2=k-j;
            if (k==n+1){
                /*if (min(num1,num2)>=a[j]) { ans+=2; break; } 
                    else { ans++; break; }
            */
                ans++; break;
            }
            else{
                int flag=0;
                if ( (a[k]-a[j]) % (k-j)==0){
                    d=(a[k]-a[j]) / (k-j);
                    if ( 1ll*a[j] - 1ll*d * (j-i) > 0 ) flag =1;
                }
                if (flag){
                    long long x=a[k];
                    for (i=k+1;i<=n;i++){
                        x+=d;
                        if (x<=0) break;
                        if (a[i]!=-1 && a[i]!=x) break;
                    }
                    ans++;
                }
                else{
                    /*if (min(num1,num2)>=a[j]) i=j+a[j];  else i=k;*/
                    ans++; i=k;
                }
            }
        }
        else
        {
            j=r[i];
            if (j==n+1) { ans++; break; } else num1=j-i-1; 
            
            int flag=0;
            if ( (a[j]-a[i]) % (j-i)==0){
                d=(a[j]-a[i]) / (j-i);
                flag =1;
            }
            if (flag){
                long long x=a[j];
                for (i=j+1;i<=n;i++){
                    x+=d;
                    if (x<=0) break;
                    if (a[i]!=-1 && a[i]!=x) break;
                }
                ans++;
            }
            else{
                i=j;
                ans++;
            }
        }
    }
    printf("%d
",ans);
}