[코딜리티 코딩테스트] 가장 작은 양의 정수 찾기

코딜리티(Codility) 코딩테스트를 풀어보았다.
https://app.codility.com/candidate-faq/ 여기에서 Take demo test 버튼을 눌러 시작 가능하다.

지문이 영어로 되어 있어서 시간이 더 걸릴 듯 하다.

This is a demo task.

Write a function:

class Solution { public int solution(int[] A); }

that, given an array A of N integers, returns the smallest positive integer (greater than 0) that does not occur in A.

For example, given A = [1, 3, 6, 4, 1, 2], the function should return 5.

Given A = [1, 2, 3], the function should return 4.

Given A = [−1, −3], the function should return 1.

Write an efficient algorithm for the following assumptions:

N is an integer within the range [1..100,000];
each element of array A is an integer within the range [−1,000,000..1,000,000].
Copyright 2009–2022 by Codility Limited. All Rights Reserved. Unauthorized copying, publication or disclosure prohibited.

Java 8, Java 11 중에 골라서 풀 수 있다.
나는 Java 8로 풀었고, Stream을 이용해 풀어보았다.

import java.util.Arrays;
import java.util.List;
import java.util.stream.Collectors;
import java.util.stream.Stream;

class Solution {
    public int solution(int[] A) {
        int answer = 1;

        List<Integer> AList = Arrays.stream(A)
            .boxed()
            .filter(n -> (n>0))
            .distinct()
            .sorted()
            .collect(Collectors.toList());

        if (AList.size() > 0) {
            for (int i : AList) {
                if (answer == i) {
                    answer++;
                } else {
                    break;
                }
            }
        }

        return answer;
    }
}

Arrays.stream(배열명).boxed() : Stream 생성
.distinct() : 중복 제거
.sorted() : 오름차순 정렬 (내림차순 .sorted(Comparator.reverseOrder()))
filter(n -> (n>0)) : 0보다 큰 값들만 필터링
.collect(Collectors.toList()) : List 형태로 반환

흠냐륑...;;

다시...

좋은 웹페이지 즐겨찾기