超现实子序列(vector存放相同数的位置,借助下标往后找)

F-超现实子序列_牛客练习赛109 (nowcoder.com)

题解:
由于ai<= 1e6我们可以利用vector存放相同每个数的下标,

在找到当前位置是这个数a[i]的第一个后开始往后找

 下标随着这个序列规则变化时,我们找下个下标是否存在,并且是否存在位置在上一个下标右面的

不断往后找即可

#include<iostream>
#include<algorithm>
#include<string>
#include<cstring>
#include<vector>
#include<map>
#include<queue>
using namespace std;
#define int long long
const int N = 1e6 + 10;
int a[N];
typedef pair<int, int> PII;
int mod = 1e9 + 7;
vector<int> p[N];
void solve() 
{
	int n;
	scanf("%lld",&n);
	for(int i = 1;i <= n;i++)
	{
		scanf("%lld",&a[i]);
		p[a[i]].push_back(i);
	}
	int ans = 0;
	for(int i = 1;i <= n;i++)
	{
		if(p[a[i]][0] == i)//由于每次找都是从当前数第一次出现的位置找 
		{                  //既保证了找的序列最长,又保证了找的次数很少,时间复杂度小 
			int l = a[i],r = a[i],now = i,d = 1;
			while(1)
			{
				int x;
				if(d == 1)
				{
					++r;
					x = r;
				}
				else
				{
					--l;
					x = l;
				}
				d ^= 1;
			if(x <= 0|| x > 1000000)
			break;
			int pos = lower_bound(p[x].begin(),p[x].end(),now) - p[x].begin();
			if(pos < p[x].size())
			{
				now = p[x][pos];
			}
			else
			{
				break;
			}
			}
			ans = max(ans,r - l);
		}
	}
	printf("%lld",ans);
	
	
}
signed main() 
{
//	ios::sync_with_stdio(0);
//	cin.tie(0);cout.tie(0);
	int t = 1;
//	cin >> t;
//scanf("%lld",&t);
	while (t--) 
	{
		solve();
	}
}
//3 F
//5 B
//6 F
//9 F
//10 B
//12 F
//15 FB
//18 FB