【JZOJ3719】K-D-Sequence

description

我们称一个数列为一个好的k-d 数列，当且仅当我们在其中加上最多k 个数之后，数列排序后为一个公差为d 的等差数列。

你手上有一个由n 个整数组成的数列a。你的任务是找到它的最长连续子串，使得满足子串为好的k-d 数列。

---

analysis

• 满足条件的([l,r])子序列中所有的数模(d)余数相同、没有相同的数，且最大值(-)最小值(≤r-l+k+1)

• (i)循环扫一遍，然后维护一个单调递增的右端点(j)，由性质可知在(i)左移的过程中右端点不会回调

• 拿一个东西维护([i,j])中间的元素的最大最小值、支持插入删除，比较好想的是哈希加线段树，查询就线段树上二分

• 或者写个(set)维护元素，时间复杂度(O(nlog n))

---

code

#pragma GCC optimize("O3")
#pragma G++ optimize("O3")
#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<set>
#define MAXN 200005
#define INF 100000000000
#define ll long long
#define reg register ll
#define fo(i,a,b) for (i=a;i<=b;++i)
#define fd(i,a,b) for (i=a;i>=b;--i)

using namespace std;

ll n,k,d,i,j,ansl,ansr;
ll a[MAXN];
set<ll>s;

int main()
{
	freopen("sequence.in","r",stdin);
	freopen("sequence.out","w",stdout);
	scanf("%lld%lld%lld",&n,&k,&d);
	//fo(i,1,n)a[i]=read()+INF;
	fo(i,1,n)scanf("%lld",&a[i]),a[i]+=INF;
	if (!d)
	{
		i=1;
		while (i<=n)
		{
			j=i;
			while (a[i]==a[j+1] && j<n)++j;
			if (j-i>ansr-ansl)ansl=i,ansr=j;
			i=j+1;
		}
		printf("%lld %lld
",ansl,ansr);
		return 0;
	}
	fo(i,1,n)
	{
		if (n-i<ansr-ansl)break;
		
		if (i<=j && j-i>=ansr-ansl)
		{
			set<ll>::iterator i1=s.begin(),i2=s.end();i2--;
			if ((*i2)-(*i1)<=j-i+1+k)ansl=i,ansr=j;
		}

		fo(j,j+1,n)
		{
			if (a[i]%d!=a[j]%d || s.count(a[j]/d))break;
			s.insert(a[j]/d);
			if (j-i>ansr-ansl)
			{
				set<ll>::iterator i1=s.begin(),i2=s.end();i2--;
				if ((*i2)-(*i1)<=j-i+1+k)ansl=i,ansr=j;
			}
		}
		s.erase(a[i]/d),--j;
	}
	printf("%lld %lld
",ansl,ansr);
	return 0;
}