2019南京网络赛题目F Greedy Sequence（暴力/set+二分 预处理 + 记忆化搜索）

题目链接

题意: 从1到n一行内输出si，si为数字i开头的满足题目要求的序列长度。

题目要求此序列满足：

1.相邻两元素在原数组中的距离不超过k。

2.该序列以字典序为第一优先级（越大越好）。

3.序列中无重复元素。

---

---

思路: 字典序为第一优先级，那么每个元素的后面跟的一定是以它为中心2k范围内<=它的最大元素。那么记忆化搜索一下。但是这个k如果很大的话，每次枚举找它的下一个元素的过程就已经是o(n^2) 的了（此处t了三发，，，哭了）。那么势必要剪枝，当时的做法是按值的大小排序，然后往后找第一个和它下标距离在k以内的元素，break。预处理出每个数的下一个元素，然后O(n)进行记忆化搜索。但是其实这种做法最坏情况也是o(n^2)的，但是没想到过了，，还挺快。后来看题解，正解是滑动窗口+set二分，写了一下发现没有暴力的快，，不知道是我写的有问题还是啥，，，

这分别是两种做法的时间，上面的是暴力，下面的是set+二分。下面贴上两种做法的代码。

做法1：暴力+贪心

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef pair<int,int> pa;
#define inf 0x3f3f3f3f
const int N = 1e5+10;
int n,k;
int a[N],dp[N],pos[N];
pa b[N];
int dfs(int x)
{
	if(dp[a[x]]!=-1)
		return dp[a[x]];
	if(!pos[x])
		return dp[a[x]]=1;
	dp[a[x]]=dfs(pos[x])+1;
	return dp[a[x]];
}
int main()
{
	int t;
	scanf("%d",&t);
	while(t--)
	{
		scanf("%d%d",&n,&k);
		for(int i=1;i<=n;i++)
		{
			scanf("%d",&a[i]);
			b[i]={a[i],i};
		}
		sort(b+1,b+n+1);
		memset(pos,0,sizeof pos);
		for(int i=n;i>=1;i--)
		{
			int x=b[i].second;
			for(int j=i-1;j>=1;j--)
			{
				if(abs(x-b[j].second)<=k)
				{
					pos[x]=b[j].second;break;
				}
			}
		}
		memset(dp,-1,sizeof dp);
		for(int i=1;i<=n;i++)
		{
			if(dp[a[i]]==-1)
				dfs(i);
		}
		for(int i=1;i<=n;i++)
		{
			if(i>1)
				printf(" ");
			printf("%d",dp[i]);
		}
		printf("\n");
	}
	return 0;
}

---

---

做法2：set+二分预处理

#include<iostream>
#include<cstdio>
#include<cstring>
#include<set>
#include<algorithm>
using namespace std;
const int N = 1e5+10;
int a[N],pos[N],jump[N],f[N];
int dfs(int x)
{
	if(f[a[x]]!=-1)
		return f[a[x]];
	int num=jump[x];
	if(!num)
		return f[a[x]]=1;
	f[a[x]]=dfs(num)+1;
	return f[a[x]]; 
}
int main()
{
	int t;
	cin>>t;
	while(t--)
	{
		int n,k;
		scanf("%d%d",&n,&k);
		for(int i=1;i<=n;i++)
		{
			scanf("%d",&a[i]);
			pos[a[i]]=i;
		}
		set<int> s;
		set<int>::iterator it;
		for(int i=1;i<=k;i++)
			s.insert(a[i]);
		s.insert(0);
		memset(f,-1,sizeof f);
		memset(jump,0,sizeof jump);
		for(int i=1;i<=n;i++)
		{
			if(i+k<=n)
				s.insert(a[i+k]);
			if(i-k>1)
				s.erase(a[i-k-1]);
			it=s.lower_bound(a[i]);
			if(it!=s.end())
				jump[i]=pos[*(--it)];
		}
		for(int i=1;i<=n;i++)
			if(f[a[i]]==-1)
				dfs(i);
		for(int i=1;i<=n;i++)
		{
			if(i>1) printf(" ");
			printf("%d",f[i]);
		}
		printf("\n");
	}
	return 0;
}