[数据结构与算法]【双指针】D. Cyclic Rotation

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

题意?
给定一个数组长度 n，并给定数组 a。
可以对 a 数组实行一个操作，选择两个下标 l ,r??,对于?? 1 ≤ l ≤ r ≤ n ??&&?? a[l] = a[r]
?将a[l]放到a[r]的后面，此时 $a[l?r]=[a_{l+1},a_{l+2},??,a_r,a_l]$
现在给出另一个数组 b。问数组 a 能否通过变换，变为数组 b 。

双指针做法：

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

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

注意：如果出现多个连续的相同的数，我们需要对$b[j]$进行重复计入，而不能只计入一次。
例如$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]$这个值可能就是跳到后面的，如果小于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;
}

创作打卡挑战赛

赢取流量/现金/CSDN周边激励大奖