Swift辞書初期化のランタイムエラー

Question

辞書のデータ更新の問題についてです。
問題URL（https://paiza.jp/works/mondai/c_rank_level_up_problems/c_rank_dictionary_step2）

問題文
n 人の人の名前 s_1, ..., s_n が与えられたのち、 m 回の「攻撃」に関する情報が与えられます。各行は “p_i a_i” というフォーマットで与えられ、 p_i はダメージを受けた人の名前 （s_1, ..., s_n のいずれか） 、 a_i は p_i が受けたダメージ数を表す数です。

最後に人名 S が与えられるので （S は s_1, ..., s_n のいずれか） 、 S が受けたダメージの合計を出力してください。なお、一度もダメージを受けていない人の合計ダメージは 0 とします。

条件
・1 ≤ n, a_i ≤ 50 （1 ≤ i ≤ n）
・各 s_i （1 ≤ i ≤ n） は大小アルファベットからなる長さ 1 以上 10 以下の文字列で、互いに全て異なる

入力例
2
Kirishima
Kyoko
2
Kyoko 1
Kyoko 2
Kyoko

出力例
3

このコードで正解はできたのですが納得できないところがあります。

var dic : [String:Int] = [:]
let n = Int(readLine()!)!

for _ in 0..<n {
    dic[readLine()!] = 0
}

let m = Int(readLine()!)!

for _ in 0..<m {
    let arr = readLine()!.split(separator: " ").map{ String($0) }
     dic[arr[0]]! += Int(arr[1])!
}
let S = readLine()!

print(dic[S]!)

---

一つ目

dic[readLine()!] = 0

は

dic = [readLine()!,0] 

だと入力によってはランタイムエラーになります。

---

２つ目

let arr = readLine()!.split(separator: " ").map{ String($0) }

はmapを使わないとエラーになりますが最初からString型ではないのでしょうか
またString型に直さないといけないのであれば

 let arr　: [String:String] = readLine()!.split(separator: " ")

でもいいのかと思いましたがダメでした。

以上の２点についてご教示いただけると幸いです。

Answer 1

一つ目

dic = [readLine()!,0]

これはコンパイルエラーになりますが、, は : の間違いでしょうか？

dic = [readLine()!: 0]

この場合、既存の辞書に要素を追加するのではなく、[readLine()!: 0] という一要素の辞書で dic 全体が上書きされます。つまり、

2
Kirishima
Kyoko
(以下略)

という入力があると、本来は ["Kirishima": 0, "Kyoko": 0] という辞書になってほしいと思いますが、まず Kirishima で dic の中身は ["Kirishima": 0] になり、次の Kyoko では ["Kyoko": 0] だけになってしまいます。

---

２つ目

String の split メソッド の戻り値の型は [Substring] で、Substring は String とは異なる型なので、[String] に変換したいなら map する必要があります。
参考: 【Swift】String.SubSequenceとは何か？ - Qiita

ちなみに、次のようにすれば arr の型は [Substring] のまま取り扱うことができます。

let arr = readLine()!.split(separator: " ")
dic[String(arr[0])]! += Int(arr[1])!

Comments

[tanakashunta]

一つ一つのメソッドについての正確な理解が大事だということ身にしみて分かりました...
とても分かりやすい回答をありがとうございます！