문제 링크 (opens new window)

문제 설명

1. 쉬운 풀이 - String 자체 메서드 활용
2. 귀찮은 풀이 - 이진수 직접 변환 (성능 안좋음)

class Solution {
    func countBits(_ n: Int) -> [Int] {
        var result: [Int] = []
        for i in 0...n {
            let str = String(i, radix: 2)
            var count = 0
            str.forEach({
                if($0 == "1") {
                    count += 1
                }
            })
            result.append(count)
        }
        return result
    }
}