AcWing 构造数组 区间合并

AcWing 构造数组 区间合并_第1张图片AcWing 构造数组 区间合并_第2张图片

思路:
这道题第一眼来看以为是动态规划类型的题目,然而尝试了用dp的方法做,然而超时了,过了差不多一半的测试店,显示的是超时。那么应该来说动态规划是可以做的,但数据卡的比较严。在看其他同学的评论后,使用KMP算法也会使得部分测试点超时。

听完讲解后,发现大佬是通过区间合并的方法实现的。

由题意可知,若a[i]等于a[j],则b[i]和b[j]也必须是相同的,而数组b是非递减的,此时数组b中第i到j项的元素必须是相同的,才能符合题目要求。所以我们要统计这样的区间的个数。运用区间合并的方法,得出最后的区间数为k。两个相邻区间之间有2种变化方式,因此总可能数是2的k-1次方个。

参考代码:

#include
#include
using namespace std;
#define INF 0x3f3f3f3f
typedef long long ll;
vector<pair<int, int>> v;

int main()
{
	ios::sync_with_stdio(false);
	cin.tie(nullptr);
	int n;
	cin >> n;
	unordered_map<int, int> l(300000), r(300000);
	int num[200005];
	for (int i = 0; i < n; i++)
	{
		cin >> num[i];
		r[num[i]] = i;
	}
	for (int i = n - 1; i >= 0; i--)
		l[num[i]] = i;
	for (auto it = l.begin(); it != l.end(); it++)
	{
		v.push_back({ it->second, r[it->first] });
	}
	sort(v.begin(), v.end());

	int cnt = 0;
	int bg = -1e9, ed = -1e9;
	for (int i = 0; i < v.size();i++)  //区间合并
	{
		if (ed < v[i].first)
		{
			if (ed != -1e9) cnt++;
			bg = v[i].first, ed = v[i].second;
		}
		else if (ed < v[i].second)
		{
			ed = v[i].second;
		}
	}
	if (bg != -1e9 && ed != -1e9) cnt++;
	int res = 1;
	for (int i = 1; i < cnt; i++)
	{
		res = res * 2 % 998244353;
	}
	cout << res;

	return 0;
}

你可能感兴趣的:(AcWing,算法)