【AtCoder】ABC361解説（Python）

デンソークリエイトプログラミングコンテスト2024（AtCoder Beginner Contest 361）の解説記事です。

ABC361 A – Insert

問題

問題文の要約は以下の通りです。

問題の要約

長さ \(N\) の整数列 \(A\) と整数 \(K\) ,\(X\) が与えられるので、整数列 \(A\) の \(K\) 要素目の直後に整数 \(X\) を挿入した整数列を出力せよ。

解説

insert関数を使用する。

解説

まず入力を受け取ります。

# 入力
N,K,X=map(int,input().split())
A=list(map(int,input().split()))

数列 \(A\) に \(X\) を insertします。 \(A\) の \(K\)番目の後に \(X\) を挿入するにはA.insert(K,X)と書きます。

# 数列AにXを挿入
A.insert(K,X)

答えを出力します。リストの要素のみ出力したいのでアスタリスク（*）をつけてアンパック (unpack) します。

# 数列を出力
print(*A)

解答

# 入力
N,K,X=map(int,input().split())
A=list(map(int,input().split()))

# 数列AにXを挿入
A.insert(K,X)

# 数列を出力
print(*A)