题目描述

小瓜现在让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;
}
Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐