1161 Merging Linked Lists

最新推荐文章于 2023-08-22 15:37:36 发布

花落的那一天

最新推荐文章于 2023-08-22 15:37:36 发布

阅读量309

点赞数

分类专栏： PAT甲级文章标签：算法数据结构

本文链接：https://blog.csdn.net/ghost_darkgreen/article/details/125417322

版权

PAT甲级专栏收录该内容

6 篇文章 0 订阅

订阅专栏

题意

给了两个链表，找到最短的，然后插入长的那个里面，长的每隔两个插一个短的，长的长度肯定是短的二倍以上。

Code

#include <iostream>
#include <vector>
#include <algorithm>
#include <iomanip>

using namespace std;

const int N = 1e6;

struct node
{
    int add;
    int data;
    int ne;
}node[N];


int main()
{
    int L1, L2, n;
    cin >> L1 >> L2 >> n;
    for (int i = 1; i <= n; i++)
    {
        int a, b, c;
        cin >> a >> b >> c;
        node[a] = {a, b, c};
    }
    vector<int> l1, l2;
    int t = L1;
    while (t != -1)
    {
        l1.push_back(t);
        t = node[t].ne;
    }
    t = L2;
    while (t != -1)
    {
        l2.push_back(t);
        t = node[t].ne;
    }
    if (l1.size() < l2.size()) l1.swap(l2);
    reverse(l2.begin(), l2.end());
    // for (int i = 0; i < l1.size(); i++) cout << l1[i] << endl;
    vector<int> res;
    int cnt = 0;
    for (int i = 0, j = 0; i < l1.size(); i++)
    {
        cnt++;
        res.push_back(l1[i]);
        if (cnt == 2 && j < l2.size()) 
        {
            res.push_back(l2[j++]);
            cnt = 0;
        }
    }
    for (int i = 0; i < res.size(); i++)
    {
        if (i + 1 == res.size())
        {
            cout << setw(5) << setfill('0') << node[res[i]].add << ' ' << node[res[i]].data << ' ' << -1 << endl;
        }
        else
        {
            cout << setw(5) << setfill('0') << node[res[i]].add << ' ' << node[res[i]].data << ' ' << setw(5) << setfill('0') << node[res[i + 1]].add << endl;
        }
    }
    return 0;
}