[1337] 올바른 배열

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

1337번: 올바른 배열

투 포인터로 접근한 문제다.

 

풀이

기준이 될 인덱스 i는 1부터 n까지, 비교할 인덱스 k는 i+1부터 i+4까지 반복문을 수행. 둘의 차이가 5보다 작다면 올바른 배열로 판정한다.

기준 인덱스 i가 바뀔 때마다 최댓값을 갱신한 뒤, 반복문이 끝나고 나면 값을 출력하면 된다.

 

정답 코드

	import Foundation
	let n = Int(readLine()!)!
	var map = [Int]()
	var ans = 0
	var cnt = 1
	for _ in 0..<n{
	map.append(Int(readLine()!)!)
	}
	map.sort(by: <)
	for i in 0..<n{
	cnt = 1
	for k in i+1..<i+5{
	if k>=n{ break }
	if map[k]-map[i]<5{ cnt += 1}
	}
	ans = max(ans, cnt)
	}
	print(5-ans)

view raw 1337.swift hosted with ❤ by GitHub