Description
被认为天才的小头遇到麻烦了!!这天数学课老师给出了一道难题,而小头居然没能在3秒内解决,可见此题难度之大。
问题是这样的:n个整数围成一个环,老师要求选出其中的若干数,使得选中的数所组成的环中,两个相邻数的差的绝对值不等于1。在满足这个前提下,问最多能取多少个数。
Input
第一行一个正整数n,表示有n个数
第二行n个整数,a1、a2……an 按顺时针方向围成一个环。
Output
一个正整数,即表示最多能选多少个数。
Sample Input
5
1 2 3 5 2
Sample Output
3
枚举完起点把环转化成一行后,用dp[i]表示以第i个数结尾最多能选多少个数,那么在求dp[i]时只需要记录dp[1]到dp[i-1]里前3大的数就可以了(3个不同数中至少有一个和a[i]的绝对值不等于1)。同理,在枚举起点时也只要枚举前5个不同的数。因此该题复杂度为O(N)。同时dp【i】 要使用线段树来维护。
代码如下,谢绝抄袭,特别是什么标哥标姐,是谁我不指明了,给你留点面子。
#include<cstdio>
#include<cstring>
#define F(i,x,y) for(int i=x;i<=y;i++)
using namespace std;
const int maxn = 1000000 + 500;
int a[maxn*2],f[maxn*2],maxa,tree[maxn * 8];
int ans=0;
int n;
int max(int x,int y){return x>y?x:y;}
int find(int x,int st,int en,int l,int r)
{
if(st>en) return(0);
if(l==st && r==en) return(tree[x]);
else
{
int mid=(l+r)/2;
if(mid>=en) return(find(2*x,st,en,l,mid));
else if(mid<st) return(find(2*x+1,st,en,mid+1,r));
else return(max(find(2*x,st,mid,l,mid),find(2*x+1,mid+1,en,mid+1,r)));
}
}
void change(int x,int l,int r,int m,int v)
{
if(l==r) tree[x]=v;
else
{
int mid=(l+r)/2;
if(m<=mid) change(2*x,l,mid,m,v);else change(2*x+1,mid+1,r,m,v);
tree[x]=max(tree[2*x],tree[2*x+1]);
}
}
void init()
{
scanf("%d",&n);
F(i,1,n)
{
scanf("%d",&a[i]);
a[i+n]=a[i]; maxa=max(maxa,a[i]);
}
}
void work()
{
int s1,s2,s3;
memset(tree,0,sizeof(tree));
memset(f,0,sizeof(f));
F(i,1,2*n)
{
s1=find(1,0,a[i]-2,0,maxa);
s2=find(1,a[i]+2,maxa,0,maxa);
s3=find(1,a[i],a[i],0,maxa);
f[i]=max(max(s1,s2),s3)+1;
ans=max(ans,f[i]);
change(1,0,maxa,a[i],f[i]);
}
}
void getout()
{
printf("%d\n",ans/2);
}
int main()
{
init();
work();
getout();
return 0;
}