2つのソート済みの整数列を併合する

Question

前提・実現したいこと

2つのソート済みの整数列を併合させる問題です。
答えは

[7,8,15,20,24,25,26,30,32]

としたいのですが、どこが違うのかがわかりません。

発生している問題・エラーメッセージ

8 15 20 26 30 32
7 24 25
[7, 24, 25]

該当のソースコード

python

a=[]
b=[]
a.append([int(i) for i in input().split()])
b.append([int(i) for i in input().split()])
d=[]
i=0
j=0

while i<len(a) and j<len(b):
if a[i]<=b[j]:
d.append(a[i])
i+=1

else:
    d.append(b[j])
    j+=1    
    

print(*d)

試したこと

補足情報（FW/ツールのバージョンなど）

Comments

[takasima20]

とりあえず a b は append じゃなくて、直接代入しましょう。

Answer 1

他の方が回答されている通りですね。
参考までに地道なやり方だとこんなかんじでしょか。

a = [int(i) for i in input().split()]
b = [int(i) for i in input().split()]

d = []
i = 0
j = 0
while i<len(a) and j<len(b):
    if a[i]<=b[j]:
        d.append(a[i])
        i += 1
    else:
        d.append(b[j])
        j += 1
while i<len(a):
    d.append(a[i])
    i += 1
while j<len(b):
    d.append(b[j])
    j += 1

print(*d)

Comments

[apaka]

みなさんの言う通りでした！ありがとうございました！

[takasima20]

(･∀･)b

Answer 2

残りを加えましょう(スライスにすることで範囲外の場合は空リストが返ります)。

a = [8, 15, 20, 26, 30, 32]
b = [7, 24, 25]

d=[]
i=0
j=0

while i < len(a) and j < len(b):
  if a[i] <= b[j]:
    d.append(a[i])
    i += 1
  else:
    d.append(b[j])
    j += 1

d += a[i:] + b[j:]

print(*d) 


#
7 8 15 20 24 25 26 30 32

Comments

[apaka]

わざわざありがとうございました！

Answer 3

処理に「片方のリストがなくなったとき」の考慮が抜けています。

片方がなくなれば、残りをすべて結果リストに入れる必要がありますね。

Comments

[apaka]

おっしゃる通りでした。勉強になりました。ありがとうございました！

Answer 4

現状のプログラムでは、a, bどちらかの要素を全て処理してしまったところでwhileループを抜けてしまいます。
ループの後に残ってしまった要素を追加する処理が必要です。
たとえば、以下のような感じで。

a = [8, 15, 20, 26, 30, 32]
b = [7, 24, 25]

d = []
i = 0
j = 0

while i < len(a) and j < len(b):
    if a[i] <= b[j]:
        d.append(a[i])
        i += 1
    else:
        d.append(b[j])
        j += 1

print(*d)

print('=' * 20)
print('aの残り:', a[i:])
print('bの残り:', b[j:])
d.extend(a[i:] if i < len(a) else b[j:])
print(*d)

Answer 5

適当にインデントを付けて実行してみると、

>>> print(*d)
7 8 15 20 24 25

となりました。

この結果が間違っているのは、whileの判定方法が間違っていることです。
ただし、whileの判定方法を変更するだけではだめで、その中のifの判定もかなり変更する必要があります。

Comments

[apaka]

無事にできました。ありがとうございました！