[Algorithm] 알고리즘 팁 정리 (Swift)

1. 배열의 모든 요소 더하기, 모두 곱하기

let arr = [1, 2, 3, 4, 5]

// 배열의 요소 모두 더한 값
let addTotal = arr.reduce(0, +) // 15

// 배열의 요소 모두 곱한 값
let mulTotal = arr.reduce(1, *) // 120

 

2. 공백으로 숫자 여러 개 입력 받은 후 정수 배열로 저장하기

// components 사용
var nums = readLine()!.components(separatedBy: " ").map { Int($0)! }
var nums = readLine()!.components(separatedBy: " ").compactMap { Int($0) }

// split 사용
var nums = readLine()!.split(separator: " ").map { Int(String($0))! }
var nums = readLine()!.split(separator: " ").compactMap { Int(String($0)) }

 

3. 여러 문자 (또는 숫자)를 공백 없이 입력 받았을 때,  문자 배열로 저장하기

let result = readLine()!.map { (value: Character) -> Character in
    value
}
print(result)

// 입력1 : 1234567
// 결과1 : ["1", "2", "3", "4", "5", "6", "7"]

// 입력2 : abcdefg
// 결과2 : ["a", "b", "c", "d", "e", "f", "g"]



// 또는 아래와 같이 쓰는 것도 가능하다.
// strArr 타입은 [String.Element]()이고, Element는 Character의 typealias.
let strArr = Array(readLine()!)

 

 

4. 입력이 들어오면 계속해서 연산하기

while let input = readLine() {
    print(input.split(separator: " ").map{ Int($0)! }.reduce(0, +))
}

 

5. 배열 원하는 대로 정렬하기

var students = ["Kofi", "Abena", "Peter", "Kweku", "Akosua"]
students.sort(by: >)
print(students)
// Prints "["Peter", "Kweku", "Kofi", "Akosua", "Abena"]"


var num = [1, 2, 3, 4, 5]
num.sort(by: >)
print(num) // [5, 4, 3, 2, 1]

var num2 = [5, 4, 3, 2, 1]
num2.sort(by: <)
print(num2) // [1, 2, 3, 4, 5]

 

6. 아스키코드 구하기

var word = Character(readLine()!).asciiValue // Swift 5 이상에서만 가능.
print(word) // 타입은 UInt8

// 입력: a
// 결과: Optional(97) => a의 아스키코드값은 97

// Swift 5 미만에서는 unicodeScalars 사용
let ch: Character = "Z"
let unicodeScalar = ch.unicodeScalars.first!.value // 90
print(unicodeScalar) // 90 출력. Z의 아스키코드값은 90


// 아스키코드를 문자로 바꾸기
let asciiCode = 65 // A의 아스키코드값
let asciiStr = Character(UnicodeScalar(asciiCode)!) // asciiStr에는 A가 저장된다.

 

 

7. 수의 모든 자릿수의 합 / 곱 구하기

let num = 12345
print(String(num).reduce(0) { $0 + Int(String($1))! }) // 1 + 2 + 3 + 4 + 5 = 15

let num = 12345
print(String(num).reduce(1) { $0 * Int(String($1))! }) // 1 * 2 * 3 * 4 * 5 = 120

 

 

8. 문자열 내에서 원하는 인덱스에 해당하는 문자만 뽑아내기

var str = "abcdef"
print(str[String.Index(utf16Offset: 0, in: str)]) // a
print(str[String.Index(utf16Offset: 1, in: str)]) // b
print(str[String.Index(utf16Offset: str.count - 1, in: str)]) // f
print(str[String.Index(utf16Offset: 1, in: str)...String.Index(utf16Offset: 3, in: str)]) // bcd
print(str[String.Index(utf16Offset: 1, in: str)...]) // bcdef
print(str[...String.Index(utf16Offset: 3, in: str)]) // abcd

 

 

9. 문자열로 이루어진 숫자를 정수로 바꾸기

Int(s)! // 부호 (+, -) 들어가도 상관없음.

 

 

10. 숫자만 찾기

var s = "12345"
var charSet = CharacterSet(charactersIn: "0123456789").inverted // 숫자를 제외한 모든 문자

if let _ = s.rangeOfCharacter(from: charSet) {
    print("숫자 아닌 문자 포함되어있음")
} else {
    print("숫자만 포함되어있음")
}

 

 

11. [String] 배열 원소를 하나로 이어서 출력하기

let str = ["H", "e", "l", "l", "o"] // [String]
print(str.joined()) // Hello

 

12. 백준에서는 문자열보다 배열을 사용하는 게 낫다. 문자열 쓰면 시간초과 나는 경우가 굉장히 많다.

배열 중에서도 [Character] 배열보다 [String] 배열을 애용하도록 하자.

 

 

13. 배열에서 가장 큰 값 (max)과 가장 작은 값 (min) 찾기

let arr = [1, 2, 3, 4, 5]
print(arr.max()!) // 5
print(arr.min()!) // 1

 

14. 배열에 enumerated()를 사용하면, (index, value) 튜플 타입으로 구현된 리스트형이 리턴된다.

var result = ["A", "B", "C", "D"]
var dictionary = [String: Int]()

// array 값에 enumerated()함수를 사용하면, (index, value) 튜플형식으로 구현된 리스트형이 리턴된다.
for (index, value) in result.enumerated() {
  dictionary[value] = index
}

print(dictionary) // ["B": 1, "A": 0, "C": 2, "D": 3]

 

15. zip 사용법

var result = ["A", "B", "C", "D"]
var anotherResult = ["💜", "💛", "💚", "💙"]
var dictionary = [String: String]()

for (alphabet, heart) in zip(result, anotherResult) {
  dictionary[alphabet] = heart
}
print(dictionary) // ["D": "💙", "C": "💚", "A": "💜", "B": "💛"]

 

16. 문자열을 String 배열 또는 Character 배열로 변환하기

// 문자열 -> String 배열
let str1 = "ABC"
let array1 = str1.map { String($0) }
print(array1, type(of: array1)) // ["A", "B", "C"] Array<String>

// 문자열 -> Character 배열
let str2 = "ABC"
let arr2 = Array(str2)
print(arr2, type(of: arr2)) // ["A", "B", "C"] Array<Character>

 

17. 입력 2개를 받아서 2차원 배열 만들기

# 예시 출력: [["사과", 1], ["포도", 2], ["복숭아", 3], ["딸기", 4], ["파인애플", 5]]
let fruitList = readLine()!.split(separator: " ").map { String($0) }
let countList = readLine()!.split(separator: " ").compactMap { Int($0) }

// zip을 이용해 이름과 나이를 튜플로 묶은 배열 생성
let pairedList = Array(zip(fruitList, countList))

print(pairedList)

 

18. 입력 2개 받아서 튜플 리스트로 만들기

# 예시 출력: [("사과", 1), ("포도", 2), ("복숭아", 3), ("딸기", 4), ("파인애플", 5)]
let fruitList = readLine()!.split(separator: " ").map { String($0) }
let countList = readLine()!.split(separator: " ").compactMap { Int($0) }

// zip을 이용해 이름과 나이를 묶은 후, 각각을 이차원 배열로 변환
let pairedList = zip(fruitList, countList).map { [$0.0, "\($0.1)"] }

print(pairedList)

 

19. 배열의 중복 요소 제거하기 (중복 제거 & 원래 배열 순서 유지)

let array = [1, 2, 2, 3, 4, 4, 5]
var seen = Set<Int>()

let uniqueArray = array.filter { element in
    if seen.contains(element) {
        return false
    } else {
        seen.insert(element)
        return true
    }
}

print(uniqueArray)  // 출력: [1, 2, 3, 4, 5]


// 축약 코드
let array = [11, 5, 125, 3, 125, 2, 2, 3, 4, 4, 5]
var seen = Set<Int>()

let uniqueArray = array.filter { seen.insert($0).inserted }

print(uniqueArray)  // 출력: [11, 5, 125, 3, 2, 4]

 

20. 특정 자릿수까지 반올림해서 나타내기

let result = Double(Double(1)/Double(6))
print(result) // 0.16666666666666666

// 반올림
let formatted = String(format: "%.5f", result)
print(formatted) // 0.16667

let multiplier = 100000.0  // 10^5, 소숫점 5자리까지라고 가정

// 올림:
let ceilingResult = Double(ceil(result * multiplier)) / multiplier
print("올림:", ceilingResult)  // 출력 예: 0.16667 (입력값에 따라 달라짐)

// 버림:
let floorResult = Double(floor(result * multiplier)) / multiplier
print("버림:", floorResult)  // 출력 예: 0.16666 (입력값에 따라 달라짐)

 

23. String에서 문자와 숫자만 남기기

var inputString = "A man, a plan, a canal: Panama"
var newString: String = ""
inputString.forEach {
    if ($0.isLetter || $0.isNumber) {
        newString.append($0)
    }
}
print(inputString)
print(newString) // AmanaplanacanalPanama