n 個の文字と数字の組をreadLine()で読み込み？同じ文字同士で数字の計算

Question

プログラミング Swift 初心者

ある問題に挑戦していてどうしても分からない問題があります。
問題
n 個の文字と数字の組をreadLine()で読み込み？同じ文字同士で数字の計算（加算）をします。そして、最後に昇順で出力します。

入力例
5
A 1
D 6
A 2
G 4
B 70

出力例
B 70
D 6
G 4
A 3

この形で出力したいです。入力も出力もpaizaというサイトのエディターで行います。

辞書型で格納して、同じ文字同士で加算みたいなことができたらと思いましたが、辞書型はkeyの重複はダメですし、配列で格納した場合、最後に出力する時に文字と数字のペアを崩さずに出力する方法が思いつきませんでした。

自分で調べることが大事だとは思いましたが、どうか教えていただきたいです。
ご回答よろしくお願いいたします。

Answer 1

少しくどく書くならこのような感じになります。

Swift の辞書には、キーとして与えられた値が存在しない場合のデフォルト値を与える方法があるので、それを活用して加算します。

• Dictionary - subscript(_:default:)

Accesses the value with the given key. If the dictionary doesn’t contain the given key, accesses the provided default value as if the key and default value existed in the dictionary.

defautl: を知らなくてもオプショナルバインディングなどで nil 判定を使ったりすれば書けなくもないですが、おそらくこちらの方がスマートにかけると思います。

• [https://paiza.io/projects/pq0a6HvtTXC-ouMDG_jfAA

](https://paiza.io/projects/pq0a6HvtTXC-ouMDG_jfAA)

import Foundation

// 読み込む行数
let n = Int(readLine()!)!

// 結果を入れる辞書
var result: [String: Int] = [:]

// 一行ずつ n 行読み込む
for _ in 0..<n {
    // 一行読み込み、空白で区切った配列にする
    let line = readLine()!.split(separator: " ")
    // 配列の戦闘要素を出出現文字とする
    let word = String(line[0])
    // 配列の 2 版めの要素をその文字に与えられた数値とする
    let num = Int(line[1])!
    
    // 辞書を使って加算する
    result[word, default: 0] += num
}

// 辞書の結果をソートし、タプルの配列にする
let sortedResult = result.sorted(by: { $0.value > $1.value })


// 整形して出力する
for line in sortedResult {
    print("(line.key) (line.value)")
}

あるいは、こういう書き方もできるかと思います。

こちらは、Dictionaryの reduce を使っています。

• reduce(into:_:)

Returns the result of combining the elements of the sequence using the given closure.

_ = readLine()

Array(AnyIterator { readLine() })
    .map { $0.split(separator: " ")
        .map { String($0) }
    }
    .reduce(into: [:]) { result, num in
        result[num[0], default:0] = Int(num[1])!
    }
    .sorted { $0.value > $1.value }
    .forEach {
        print($0.key, $0.value)
    }

• https://paiza.io/projects/WW0mdYV4Npq0UUPT94QTbw

---

ちなみに、最初のコードを default: を使わずに書くとこのような感じになります。

import Foundation

// 読み込む行数
let n = Int(readLine()!)!

// 結果を入れる辞書
var result: [String: Int] = [:]

// 一行ずつ n 行読み込む
for _ in 0..<n {
    // 一行読み込み、空白で区切った配列にする
    let line = readLine()!.split(separator: " ")
    // 配列の戦闘要素を出出現文字とする
    let word = String(line[0])
    // 配列の 2 版めの要素をその文字に与えられた数値とする
    let num = Int(line[1])!
    
    if result[word] == nil {
        // 初回の登録時は値を代入
        // ここはアンラップしない
        result[word] = num
    } else {
        // 初回以降は値を加算する
        // こちらはアンラップする
        result[word]! += num
    }
}

// 辞書の結果をソートし、タプルの配列にする
let sortedResult = result.sorted(by: { $0.value > $1.value })


// 整形して出力する
for line in sortedResult {
    print("(line.key) (line.value)")
}

Dictionary に値を代入するときにはアンラップせず、一方加算するときにはアンラップするなど、辞書の振る舞いに気をつけなければいけない分、多少ややこしくなります。

Comments

[anmoanmo]

初回の登録時に値を代入といった発想ができませんでした。
大変勉強になりました。
ありがとうございました！！