0
点赞
收藏
分享

微信扫一扫

【双指针】D. Cyclic Rotation

DYBOY 2022-04-25 阅读 36

题目链接:
https://codeforces.com/contest/1672/problem/D


双指针做法:

操作是对于a数组两端相同的数进行操作,把左边的数放到右边

我们可以倒着想,对于b数组,如果 b [ j ] = = b [ j + 1 ] b[j] == b[j +1] b[j]==b[j+1],那么 b [ j ] b[j] b[j]就有可能是从j位置前面来的,我们对这个 b [ j ] b[j] b[j]进行计数。

注意:如果出现多个连续的相同的数,我们需要对 b [ j ] b[j] b[j]进行重复计入,而不能只计入一次。
例如 a = [ 3 , 3 , 4 , 3 ] , b = [ 4 , 3 , 3 , 3 ] a = [3,3,4,3],b=[4,3,3,3] a=[3,3,4,3],b=[4,3,3,3]

cnt[i]代表的是i这个数可能从前面来的次数
i代表的是对数组a的遍历
j代表的是当前b数组遍历到的位置,如果b数组中出现连续的元素,那么j一定会在这段连续元素的第一个元素的位置(对应代码中的while语句)

对于每次循环,判断a[i]a[j]是否相等

  • 如果相等的话,说明当前位值可以匹配,对应j--
  • 不相等的话,判断是否cnt[a[i]]大于0,如果大于0,说明 a [ i ] a[i] a[i]这个值可能就是跳到后面的,如果小于0的话,就一定不能构造出来了,直接输出NO
#include<bits/stdc++.h>
using namespace std;
using ll = long long;
const int N = 2e5 + 5, mod = 998244353;

void solve()
{
	int n;
	cin >> n;
	vector<int> a(n + 1), b(a);
	for(int i = 1; i <= n; i++)
		cin >> a[i];
	for(int i = 1; i <= n; i++)
		cin >> b[i];

	vector<int> cnt(n + 1);
	for(int i = n, j = n; i >= 1; i--)
	{
		while(j > 1 and b[j] == b[j - 1]) ++cnt[b[j]], --j;

		if(j >= 1 and a[i] == b[j]) --j;
		else if(cnt[a[i]]) --cnt[a[i]];
		else
		{
			cout << "NO\n";
			return;
		}
	}
	cout << "YES\n";
}
int main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);
	int t;
	cin >> t;
	// t = 1;
	while(t--)
		solve();
	return 0;
}
举报

相关推荐

0 条评论