【51nod 3056】【STL】小明爱数列

51nod 3056 小明爱数列
STL专题练

---

题目

小明非常喜欢数列，于是他提出了一个关于数列的问题。他拿到一个长度为n的数列，之后对这个数列进行了两种操作：

第一种：D x

表示删除数列第x位置的一个数字，后面的数字向前移动（若x>此时数列长度则操作无效）；

第二种：Z x y

表示在数列的第x位置插入一个数字y，后面的数字向后移动（若x>此时数列长度则在数列末尾插入y）；

经过k次这两种操作以后，小明想要知道现在的数列是什么，聪明的你可以帮助小明解决这个问题吗？

输入
输入包括k+2行数据；
第一行包括两个数n,k，分别表示数列中数的个数以及操作的次数，其中1=<k<=n<5000；
第二行n个数表示该数列的n个数，以空格隔开，其中任意数ai为int范围内的整数；
第三行到第k+2行，每行一个操作，格式见题面描述。
输出
输出包括两行；
第一行一个数m，表示全部操作之后数组的长度；
第二行m个数，表示当前的数组。
数据范围
对于10%的数据，1 <= n <= 4；
对于50%的数据，1 <= n <= 1024；
对于100%的数据，1 <= k<= n <=5000，且ai,x,y是int范围内的整数。
输入样例

4 3
4 5 2 1
Z 3 3
Z 0 1
D 2

输出样例

5
1 4 2 3 1

样例解释
初始数组为(4 5 2 1)
执行完Z 3 3得到(4 5 2 3 1)
执行完Z 0 1得到(1 4 5 2 3 1)
执行完D 2得到(1 4 2 3 1)

---

解题思路

就是把基本操作加上去就好了
但是如果只按题目意思打是会TLE的，因为vector的删除之类的操作是要遍历整个数组的

1. 当删除的是不存在的位，返回
2. 当加入的是最新的一位，不用insert()，用push_back()

---

Code

#include <bits/stdc++.h>

using namespace std;

vector<int> v;
int n, m, x, y, len;

int main() {
	scanf("%d %d", &n, &m);
	for(int i = 0; i < n; i ++) {
		scanf("%d", &x);
		v.push_back(x);
	}
	len = n;
	for(int i = 1; i <= m; i ++) {
		char c; cin >> c;
		if(c == 'D') {
			scanf("%d", &x);
			if(x >= v.size()) continue;
			v.erase(v.begin() + x);
			len --;
		} else {
			scanf("%d %d", &x, &y);
			if(x >= v.size())
				v.push_back(y);
			else v.insert(v.begin() + x, y);
			len ++;
		}
	}
	printf("%d\n", len);
	vector<int>::iterator it;
	for(it = v.begin(); it != v.end(); it ++)
		printf("%d ", *it);
}