东华OJ 进阶题41 盾神与条状项链

问题描述 :

有一天，盾神捡到了好多好多五颜六色的珠子！他心想这些珠子这么漂亮，可以做成一条项链然后送给他心仪的女生~于是他用其中一些珠子做成了长度为n的项链。当他准备把项链首尾相接的时候，土方进来了。
　　“哇这么恶心的项链你也做得出来！！！”
　　盾神自知审美不是他的长项，于是他很谦虚地请教土方，怎么才能把项链做得漂亮。
　　“这个嘛，首先你要在这里加上一个这种颜色的珠子，然后在这里去掉这个珠子，然后……，最后你看看是不是漂亮很多咧~”土方一下子说出了m个修改步骤。
　　盾神觉得这个用人工做太麻烦了，于是交给了你。

输入说明 :

第一行两个数，分别为n，m。
　　第二行n个数，表示盾神一开始的项链。第i个数表示第i颗珠子的颜色。
　　接下来m行，为以下形式之一：
　　ADD P Q：表示在颜色为P的珠子前面加上一个颜色为Q的珠子。
　　DEL P：表示把颜色为P的珠子去掉，如果它不在端点处，则需要把它旁边的两颗珠子连起来。例如某时刻项链状态为1 4 5 8，则执行DEL 4会变成1 5 8，执行DEL 1会变成4 5 8。
　　输入保证在每次操作之前，项链有颜色为P的珠子，且任意时刻珠子颜色互不相同。

  表示颜色的数字不超过10^5的正数，1<=n<=10^4，1<=m<=10^4。

输出说明 :

第一行为一个数len，为做完所有操作后，项链的长度。
　　第二行len个数，表示此时项链的状态。第i个数表示第i颗珠子的颜色。

输入范例 :

10 5
1 2 3 4 5 6 7 8 9 10
DEL 5
ADD 7 5
DEL 10
ADD 4 20
ADD 20 12

输出范例 :

11
1 2 3 12 20 4 6 5 7 8 9 

完整C++代码

本来想用find函数【find(v.begin(), v.end(), t1)】，但是好像用erase删除后，find还是会返回原始数组中t1的位置，如果有知道的麻烦私信或者评论区交流🥰

#include <bits/stdc++.h>
using namespace std;

typedef vector<int> vec;

int n, m, t1, t2;
string s;
vec v;

// 循环遍历寻找 
int findNum(vec v, int num) {
	for (int i = 0; i < v.size(); i++) {
		if (num == v[i]) return i;
	}
	return -1;
}

int main() {
	cin >> n >> m;
	while (n--) {
		cin >> t1;
		v.push_back(t1);
	}
	while (m--) {
		cin >> s >> t1;
		// 寻找坐标 
		int idx = findNum(v, t1);
		// 两种操作 
		if (s[0] == 'D') {
			// 删除 
			v.erase(v.begin() + idx);
		} else {
			cin >> t2;
			// 添加 
			v.insert(v.begin() + idx, t2);
		}
	}
	cout << v.size() << endl;
	for (int i = 0; i < v.size(); i++) {
		cout << v[i] << " ";
	}
	return 0;
}

😋欢迎大伙私信或者评论区交流讨论😋