프로그래머스 level1 : 로또의 최고 순위와 최저 순위 - JAVA

 

 

코딩테스트 연습 - 로또의 최고 순위와 최저 순위 | 프로그래머스 (programmers.co.kr)

코딩테스트 연습 - 로또의 최고 순위와 최저 순위

 

1. IntStream

import java.util.stream.IntStream;

class Solution {
    public static int rank(int num) {
        if (num >= 2)
            return 7 - num;
        return 6;
    }
    
    public int[] solution(int[] lottos, int[] win_nums) {
        int zeroCount = 0;
        int correctCount = 0;
        for (int lotto : lottos) {
            if (lotto == 0) {
                zeroCount += 1;
            } else {
                if (IntStream.of(win_nums).anyMatch(x -> x == lotto))
                    correctCount += 1;
            }
        }
        int[] answer = {rank(zeroCount + correctCount), rank(correctCount)};
        return answer;
    }
}

 

2. LongStream, 람다 표현식

import java.util.Arrays;
import java.util.stream.LongStream;

class Solution {
    public int[] solution(int[] lottos, int[] winNums) {
        return LongStream.of(
                (lottos.length + 1) - Arrays.stream(lottos).filter(l -> Arrays.stream(winNums).anyMatch(w -> w == l) || l == 0).count(),
                (lottos.length + 1) - Arrays.stream(lottos).filter(l -> Arrays.stream(winNums).anyMatch(w -> w == l)).count()
        )
                .mapToInt(op -> (int) (op > 6 ? op - 1 : op))
                .toArray();
    }
}