[브론즈3][Swift]백준10810(Array)

https://www.acmicpc.net/problem/10810

 

답안코드

// N(바구니 개수)과 M(공을 넣는 횟수) 입력 받기
let input = readLine()!.split(separator: " ").map { Int($0)! }
let N = input[0]
let M = input[1]

// 바구니 초기화 (1부터 N까지의 바구니, 초기에는 모두 비어있음)
var baskets = Array(repeating: 0, count: N)

// M번 공 넣기
for _ in 0..<M {
    let operation = readLine()!.split(separator: " ").map { Int($0)! }
    let i = operation[0] - 1  // 시작 바구니 (0-based index로 변환)
    let j = operation[1] - 1  // 끝 바구니 (0-based index로 변환)
    let k = operation[2]      // 넣을 공의 번호
    
    // i번째 바구니부터 j번째 바구니까지 k번 공을 넣음
    for basket in i...j {
        baskets[basket] = k
    }
}

// 결과 출력
print(baskets.map { String($0) }.joined(separator: " "))

• N(바구니 개수)과 M(공을 넣는 횟수)을 입력받습니다.
• Array(repeating: 0, count: N)으로 N개의 바구니를 0으로 초기화합니다. 여기서 0은 바구니가 비어있음을 의미합니다.
• M번 반복하면서 다음 작업을 수행합니다:
  • i, j, k를 입력받습니다. (i: 시작 바구니, j: 끝 바구니, k: 넣을 공의 번호)
  • i부터 j까지의 바구니에 k번 공을 넣습니다. (기존에 있던 공은 빼고 새로운 공을 넣는 것입니다)
• 최종적으로 각 바구니에 들어있는 공의 번호를 출력합니다.
  • map { String($0) }으로 각 정수를 문자열로 변환합니다.
  • joined(separator: " ")로 공백을 사이에 두고 모든 문자열을 연결합니다.

주의할 점은 문제에서는 바구니와 공의 번호가 1부터 시작하지만, 배열의 인덱스는 0부터 시작하므로 입력받은 i와 j에서 1을 빼주어야 합니다.