私はSwiftにおける競プロの記事をいくつか投稿しています。
- 競プロで使える標準入力の便利メソッド一覧
- 競プロで使える標準エラーの出力方法
- 競プロで使える便利なエクステンション一覧 ←イマココ
- 競プロで使えるアルゴリズム関数一覧
自分で武器を増やしていくのが競プロの面白さのひとつだと思っているので、まずは自分で考えるのがオススメです。
どうしてもわからないときに、私の記事が参考になると嬉しいです 
はじめに
本記事は Swift/Kotlin愛好会 Advent Calendar 2020 の11日目の記事です。
空いているので埋めました。
競プロで使える便利な拡張メソッドとコンピューテッドプロパティを紹介します。
注意
- 本記事で紹介しているソースコードが正しいかは保証できていません。
- 本記事は随時更新予定です。
環境
- OS：macOS Big Sur 11.1
- Xcode：12.3 (12C33)
- Swift：5.3.2
5.2.1（2021/01/24現在のAtCoder）でも動作する
競プロで使える便利なエクステンション一覧
桁分割
説明
数値の桁を分割し、桁数の降順に整列した配列で返します。
ソースコード
private extension Numeric where Self: LosslessStringConvertible {
var digits: [Int] { string.digits }
}
private extension LosslessStringConvertible {
var string: String { String(self) }
}
private extension StringProtocol {
var digits: [Int] { compactMap { $0.wholeNumberValue } }
}
出力例
1.digits -> [1]
32.digits -> [3, 2]
445.digits -> [4, 4, 5]
62096.digits -> [6, 2, 0, 9, 6]
合計
説明
配列などシーケンスの要素の合計を返します。
ソースコード
private extension Sequence where Element: AdditiveArithmetic {
func sum() -> Element {
reduce(.zero, +)
}
}
出力例
[1].sum() -> 1
[1, 3].sum() -> 4
[1, 3, 4].sum() -> 8
(1...5).sum() -> 15
Decimal→Int変換
説明
Decimal を Int に変換します。
私は pow() の戻り値を Int で欲しいときに使います。
ソースコード
import Foundation
private extension Decimal {
var intValue: Int { NSDecimalNumber(decimal: self).intValue }
}
出力例
pow(2, 3).intValue -> 8
pow(10, 6).intValue -> 1000000
10進数→2進数変換
説明
10進数の数値を2進数の文字列に変換します。
digit に必要十分な大きさの値を渡さないとオーバーフローするので注意です。
まだAtCoderでは使ったことがありません。
ソースコード
private extension FixedWidthInteger {
func binaryString(digit: Int) -> String {
var result: [String] = []
for i in 0..<(Self.bitWidth / 8) {
let byte = UInt8(truncatingIfNeeded: self >> (i * 8))
let byteString = String(byte, radix: 2)
let padding = String(repeating: "0", count: 8 - byteString.count)
result.append(padding + byteString)
}
return String(result.reversed().joined().suffix(digit))
}
}
出力例
0.binaryString(digit: 2) -> "00"
1.binaryString(digit: 2) -> "01"
2.binaryString(digit: 4) -> "0010"
8.binaryString(digit: 8) -> "00001000"
255.binaryString(digit: 8) -> "11111111"
256.binaryString(digit: 8) -> "00000000" // !!!: オーバーフロー
256.binaryString(digit: 12) -> "000100000000"
参考リンク
素数判定
説明
整数が素数か判定します。
まだAtCoderでは使ったことがありません。
ソースコード
private extension Int {
var isPrime: Bool {
guard self > 1 else {
return false
}
return (2..<self).lazy.filter { self % $0 == 0 } .count == 0
}
}
出力例
1.isPrime -> false
2.isPrime -> true
3.isPrime -> true
4.isPrime -> false
5.isPrime -> true
回文判定
説明
整数または文字列が回文か判定します。
まだAtCoderでは使ったことがありません。
ソースコード
private extension Int {
var isPalindrome: Bool { String(self).isPalindrome }
}
private extension String {
var isPalindrome: Bool { self == String(self.reversed()) }
}
出力例
1.isPalindrome -> true
2.isPalindrome -> true
10.isPalindrome -> false
11.isPalindrome -> true
100.isPalindrome -> false
101.isPalindrome -> true
おわりに
以上、 Swift/Kotlin愛好会 Advent Calendar 2020 の11日目の記事でした。