CF 602B Approximating a Constant Range

#include<iostream>
#include<cstdio>
#include<cmath>
#include<algorithm>
#include<cstring>
#include<string>
#include<cstdlib>
#include<vector>
#include<stack>
#include<map>
using namespace std;
typedef long long ll;

int main()
{
int n,a1,a,dist1=0,dist2=0,l=1,r=1,ans=0;
scanf("%d",&n);
scanf("%d",&a1);
for(int i=2;i<=n;i++)
{
scanf("%d",&a);
dist1=a-a1; //根据题意：差值只能为0，1，-1
a1=a;

if(dist1==0) continue; //差值为0的继续往前走
if(dist1!=dist2) //差值不为0且与前一个差值不相等
{
dist2=dist1;  //更新当前区间的差值
r=i;
}
else //差值不为0但与前一个差值相同（如子序列1 2 3）
{
ans=max(ans,i-l); //更新区间长度
l=r;r=i;
}
}
ans=max(ans,n-l+1);
//若子序列的首尾相同（如子序列2 3 3 2），则ans=n-l+1；
printf("%d\n",ans);
return 0;
}