Insert 1, Insert 2, Insert 3, ... 2023牛客暑期多校训练营8 H

登录—专业IT笔试面试备考平台_牛客网

题目大意：给出一个长度为n的数组a，问有多少子串满足其可以用多个排列穿插构成

1<=n<=1e6

思路：因为每个排列的起点都是1，所以我们大致的策略就是对于每一个1，记录它往右最远到哪里可以构成合法的区间，因为1越多越好，1的数量是少于2的数量就不合法了，且我们发现越靠左的1，可能构成的区间就越长，所以如果出现不合法的情况，一定是最右边的1不合法，且它之后无论如何也不会变成合法的。

然后要判断不合法的情况，也就是a[i]的数量>a[i-1]的数量的情况，我们可以开n个栈，维护每个数字最后出现的位置，当出现a[i]的数量大于a[i-1]的数量时，最后的合法位置就是a[i-1]最后出现的位置，在那个位置右边到当前位置的1都不会再合法了，所以要单独开栈记录所有1的位置，如果出现不合法的情况，就将最后合法位置以外的1弹出，每遍历一个数就统计左边有多少个合法的1即可

//#include<__msvc_all_public_headers.hpp>
#include<bits/stdc++.h>
using namespace std;
const int N = 1e6 + 5;
typedef long long ll;
vector<int>st[N];
int a[N];
vector<int>pos;
int main()
{
	int n;
	cin >> n;
	for (int i = 1; i <= n; i++)
	{
		cin >> a[i];
	}
	ll ans = 0;
	for (int i = 1; i <= n; i++)
	{
		if (a[i] == 1)
		{
			pos.push_back(i);//记录每个1的位置
			st[1].push_back(i);//记录每个数字出现的位置
		}
		else
		{
			int r = 0;
			if (!st[a[i] - 1].empty())
			{
				r = st[a[i] - 1].back();//最后一次出现a[i]-1的位置
				st[a[i] - 1].pop_back();//维护可用的数字数量
				st[a[i]].push_back(i);
			}
			while (!pos.empty() && pos.back() > r)
			{//最后一次合法位置右边的1都不合法
				pos.pop_back();
			}			
		}
		ans += pos.size();
	}
	cout << ans << endl;
	return 0;
}