51Nod 2159 队列复原2 c/c++题解

题目描述

小瓜现在让1到n这n个整数排成一列，但是他只告诉你每个整数的后面那个数是什么（最后一个整数的后面那个数是0）。此外，他还打算在这个队列中插入m个整数，他将告诉你这m个整数插入的位置。请你帮忙复原插入m个整数之后的队列。
输入
第一行两个整数n（n<=100000）和m（m<=100），表示有n个整数，后续又将插入m个整数。
接下来n行，每行两个数i，j，表示排在整数i后面的那个数是j。
接下来m行，每行两个数a，b，表示在编号为a的数后面插入编号为b的数（保证
n+1 <= b <= n+m）
输出
n+m行，表示复原后的队列。
输入样例
4 2
1 3
3 4
4 2
2 0
1 5
2 6
输出样例
1
5
3
4
2
6

题解：
本身这道题是不难的(因为放在初学者2级题中)，但是我还是感觉做的不好，主要是前半个部分(形成n个数的序列)。
这是我的思路：
第一部分(其实也对应了队列复原1)：输入n个数对

1. 数组pre，pre[j] = i：表示i元素的前面是j，首先输入n个数对，记录下每个数的前一个数是什么。
2. 然后得到序列的最后一个元素t = pre[0]，向前遍历即可得到初始的n个数的序列sq1。
3. 得到sq1之后，我是把数据放到了sq2(vector)，虽然都是数组，但是后面涉及到元素的插入，而vector又正好有insert()函数(该函数的使用方法是：传入一个迭代器+一个数，将这个数插入到这个迭代器的前面)。
   第二部分：输入m种插入的情况。
4. 首先在sq2中找到指向a的迭代器，因为要在a的后面插入b，而直接使用insert(it,b)会将b插入到a的前面，所以可以写成insert(++it,b)，这样就可以将b插入到a的后面了，还有注意一次插入完成后，要break出循环，再进行下一次的插入操作。
5. 完成插入操作后，直接打印出sq2的元素即可。

代码：

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <climits>
#include <cstring>
#include <string>
#include <algorithm>
#include <vector>
#include <deque>
#include <list>
#include <utility>
#include <set>
#include <map>
#include <stack>
#include <queue>
#include <bitset>
#include <iterator>
using namespace std;

typedef long long ll;
const int inf = 0x3f3f3f3f;
const ll  INF = 0x3f3f3f3f3f3f3f3f;
const double PI = acos(-1.0);
const double E = exp(1.0);
const int MOD = 1e9+7;
const int MAX = 1e5+5;
int n,m;
int i,j;
int a,b;
int pre[MAX];
int sq1[MAX];
vector <int> sq2;

int main()
{
    /*
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    */

    while(cin >> n >> m)
    {
        for(int k = 1; k <= n; k++)
        {
            cin >> i >> j;
            pre[j] = i;
        }
        int cnt = 0;
        int t = pre[0];// 从后往前生成序列
        sq1[n] = t; cnt++;
        while(cnt < n)
        {
            t = pre[t];
            sq1[n-cnt] = t; cnt++;
        }
        for(int i = 1; i <= n; i++)
        {
            sq2.push_back(sq1[i]);
        }

        for(int i = 1; i <= m; i++)
        {
            cin >> a >> b;
            for(vector<int>::iterator it = sq2.begin(); it != sq2.end(); it++)
            {
                if((*it) == a)
                {
                    sq2.insert(++it,b);
                    break;
                }
            }
        }
        for(int i = 0; i < (int)sq2.size(); i++)
        {
            cout << sq2[i] << endl;
        }

    }

    return 0;
}