프로그래머스 다음 큰 숫자 Swift

문제 설명

자연수 n이 주어졌을 때, n의 다음 큰 숫자는 다음과 같이 정의 합니다.

• 조건 1. n의 다음 큰 숫자는 n보다 큰 자연수 입니다.
• 조건 2. n의 다음 큰 숫자와 n은 2진수로 변환했을 때 1의 갯수가 같습니다.
• 조건 3. n의 다음 큰 숫자는 조건 1, 2를 만족하는 수 중 가장 작은 수 입니다.

예를 들어서 78(1001110)의 다음 큰 숫자는 83(1010011)입니다.

자연수 n이 매개변수로 주어질 때, n의 다음 큰 숫자를 return 하는 solution 함수를 완성해주세요.

제한 사항

• n은 1,000,000 이하의 자연수 입니다.

입출력 예

n	result
78	83
15	23

 

풀이: 

조건 1에 부합하기 위해 answer라는 변수를 n+1로 만들어줍니다. -> var answer:Int = n+1

그리곤 while문을 사용해 answer를 조건에 부합하지 않다면 +1을 해줍니다. ->  while true{ ... answer += 1 }

조건 2에 부합하기 위해 n의 2진수로 변환했을 때 1의 갯수를 알아냅니다. 우선 String(n,radix:2)를 통해 2진수로 바꿔준 뒤 map으로 각 문자들을 나눠줍니다. 그리곤 filter를 사용해서 1만 걸러낸 뒤 count를 사용하여 그것의 갯수를 알아냅니다, -> let a = String(n, radix: 2).map{$0}.filter{$0 == "1"}.count

또한 answer도 똑같이 2진수로 변환해 1의 갯수를 알아냅니다. -> let b = String(answer, radix: 2).map{$0}.filter{$0 == "1"}.count

조건 3에 부합하기 위해서 a와 b를 비교해서 만약 같다면 while문을 바로 탈출하게 만듭니다.. -> if a == b{    break  } 

그리고 answer를 반환해주면 조건 1,2,3에 모두 충족되는 답을 반환할 수 있습니다. ->  return answer

 

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16

import Foundation   func solution(_ n:Int) -> Int {     var answer:Int = n+1     while true{         let a = String(n, radix: 2).map{$0}.filter{$0 == "1"}.count         let b = String(answer, radix: 2).map{$0}.filter{$0 == "1"}.count         if a == b{             break         }         answer += 1     }          return answer } Colored by Color Scripter

 

다른 사람의 풀이:

nonzeroBitCount라는 것을 사용해주면 이진수에서 1의 갯수를 확인할 수 있다.!!

1 2 3 4 5 6 7 8 9 10 11 12 13 14

import Foundation   func solution(_ n:Int) -> Int {     var answer : Int = n + 1       while true {         if n.nonzeroBitCount == answer.nonzeroBitCount {             break;         }         answer += 1     }       return answer } Colored by Color Scripter