Codeforces Round 946 (Div. 3) C D

比赛链接：Dashboard - Codeforces Round 946 (Div. 3) - Codeforces

写在前面:比赛时C题卡了挺久，还没做出来（qwq），D题也只是草草看了几眼，看榜单D题过的人数大于C题，写不出C题导致没心思看D题了，赛后重新补了这两道题。

---

回归正题

C题：题目链接 Problem - C - Codeforces

题意：

给定一个长度为 n 的数列 a。考虑 n−2 个关于 a 的三元对 [𝑎𝑗,𝑎𝑗+1,𝑎𝑗+2]。我们定义两个三元对是美丽的，当且仅当他们只有一个位置不同。

求这 n−2 个三元对中，美丽的三元对数量。

思路：用map映射，首先我们可以定义3个map用来存放所有三元组中的两个，然后再存一下每个三元组本身的个数，遍历数列时当出现相同的三元组中的两个，加上它的个数，当三元组本身重复出现时，需要减去它个数乘以3，最后输出ans即可。

实现代码：

#include<bits/stdc++.h>
#define int long long 
#define endl "\n"
#define fi first
#define se second
#define PII pair<int,int> 
using namespace std;
const int N=2e5+5;
int a[N];
void solve(){
	int n;
	cin >> n;
	int ans=0;
	for(int i=1;i<=n;++i) cin >> a[i];
	map<PII,int> m1,m2,m3;//存放三元组个数中的2个
	map<vector<int>,int> m;//存入三元组本身
	for(int i=1;i<=n-2;++i){
		ans+=m1[{a[i],a[i+1]}]++;
		ans+=m2[{a[i],a[i+2]}]++;
		ans+=m3[{a[i+1],a[i+2]}]++;
	    ans-=m[{a[i],a[i+1],a[i+2]}]*3;
	    m[{a[i],a[i+1],a[i+2]}]++;
	}
	cout << ans << endl;
	return ;
}
signed main(){
	ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
	int t=1;
	cin >> t;
	while(t--) solve();
	return 0;
}

D题：题目链接Problem - D - Codeforces

题意:题目给出上下左右4种操作，并对应“NSWE”，直升机和小车初始位置都在（0，0），判断他们在经过n次操作后，能否到达同一点，并且直升机和小车都至少移动一次。

思路：分析题目不难得出，我们设最后的坐标（x，y），当x或者y为奇数时，他们不可能到达同一点，否则我们可以将他们移动到（x/2,y/2）的位置上。

相当于小车和直升机每次移动每种操作数的一半

特判:

当n为2时，判断s[0]和s[1]是否相等

当n为4且操作包含“NSWE”4种操作,直升机或者小车只移动x轴或者y轴

实现代码：

#include<bits/stdc++.h>
#define int long long 
#define endl "\n"
#define fi first
#define se second
#define PII pair<int,int> 
using namespace std;
const int N=1e6+5;
int a[N];
void solve(){
	int n;
	string s;
	cin >> n >> s;
	map<char,int> m;
	int x=0,y=0;
	for(int i=0;i<n;++i){//确定（x，y）的位置，同时统计每种操作的数量
		if(s[i]=='N') y++,m[s[i]]++;
		else if(s[i]=='S') y--,m[s[i]]++;
		else if(s[i]=='E') x++,m[s[i]]++;
		else x--,m[s[i]]++;
	}
	if((x&1)||(y&1)||n==2&&s[0]!=s[1]){//不能到达同一位置输出NO
		cout << "NO" << endl;
		return ;
	}
	if(m.size()==4&&n==4){//特判
		for(int i=0;i<n;++i){
			if(s[i]=='N'||s[i]=='S') cout << 'R';
			else cout << 'H';
		}
		cout << endl;
		return ;
	}
	map<char,int> m1;//统计直升机操作的数量
	for(int i=0;i<n;++i){
		if(m1[s[i]]<m[s[i]]/2){//当直升机操作数小于总操作数/2时
			cout << 'R';//输出直升机
			m1[s[i]]++;//直升机操作的数量加1
		} 
		else cout << 'H';
	}
	cout << endl;
	return ;
}
signed main(){
	ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
	int t=1;
	cin >> t;
	while(t--) solve();
	return 0;
}