PTA 水题 两个有序链表序列的合并

7-51 两个有序链表序列的合并（20 分）

已知两个非降序链表序列S1与S2，设计函数构造出S1与S2的并集新非降序链表S3。

输入格式:

输入分两行，分别在每行给出由若干个正整数构成的非降序序列，用−1表示序列的结尾（−1不属于这个序列）。数字用空格间隔。

输出格式:

在一行中输出合并后新的非降序链表，数字间用空格分开，结尾不能有多余空格；若新链表为空，输出NULL。

输入样例:

1 3 5 -1
2 4 6 8 10 -1

输出样例:

1 2 3 4 5 6 8 10

水题

注意数据规模，以及两个空的时候输出NULL

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <string>
#include <vector>
#include <sstream>
#include <map>
#include <queue>
#include <stack>
#include <cstring>
#include <set>
#include <cmath>
#define INF 1000000

using namespace std;
int a[1000005];
int b[1000005];
int ans[2000005];
int lena;
int lenb;


int main()
{
    int i=0;
    int x=0;
    while(1)
    {
        scanf("%d",&x);
        if(x==-1) break;
        a[i++]=x;
    }
    lena=i;
    i=0;
    while(1)
    {
        scanf("%d",&x);
        if(x==-1) break;
        b[i++]=x;
    }
    lenb=i;
    int j=0;
    i=0;
    int k=0;
    if(lena==0 && lenb==0)
    {
        printf("NULL\n");
        return 0;
    }

    while(i<lena && j<lenb)
    {
        if(a[i]<b[j])
        {
            ans[k++]=a[i++];
        }
        else
        {
            ans[k++]=b[j++];
        }
    }
    while(i<lena)
    {
        ans[k++]=a[i++];
    }
    while(j<lenb)
    {
        ans[k++]=b[j++];
    }
    for(int i=0;i<k;i++)
    {
        if(i==0) printf("%d",ans[i]);
        else printf(" %d",ans[i]);
    }
    printf("\n");


    return 0;
}