Stream 연습 문제 1 - 기본

Java의 Stream 공부를 하다, 풀어볼 수 있는 연습문제가 있으면 좋겠다는 생각을 해서 포스팅 해봤습니다.
문제 난이도는 기본, 심화로 나눴습니다.
기본 문제는 모던 자바 인 액션 이라는 책의 Chapter5를 약간 변형해서 첨부하였고,
심화 문제는 알고리즘 문제를 풀면서 자주 사용할 것 같은 기법에 대해 직접 문제를 만들어 봤습니다.

혹시라도 더 효율적인 코드가 있거나 문제에 오류가 있으면 많은 피드백 부탁드립니다.

준비 코드

public class Trader {

    private final String name;
    private final String CITY;

    public Trader(String name, String CITY) {
        this.name = name;
        this.CITY = CITY;
    }

    public String getName() {
        return name;
    }

    public String getCITY() {
        return CITY;
    }

    @Override
    public String toString() {
        return "Trader{" +
            "name='" + name + '\'' +
            ", CITY='" + CITY + '\'' +
            '}';
    }
}

public class Transaction {

    private final Trader trader;
    private final int year;
    private final int value;

    public Transaction(Trader trader, int year, int value) {
        this.trader = trader;
        this.year = year;
        this.value = value;
    }

    public Trader getTrader() {
        return trader;
    }

    public int getYear() {
        return year;
    }

    public int getValue() {
        return value;
    }
    
    @Override
    public String toString() {
        return "Transaction{" +
            "trader=" + trader +
            ", year=" + year +
            ", value=" + value +
            '}';
    }
}

public class Basic {
    public static void main(String[] args) {

        Trader raoul = new Trader("Raoul", "Cambridge");
        Trader mario = new Trader("Mario", "Milan");
        Trader alan = new Trader("Alan", "Cambridge");
        Trader brian = new Trader("Brian", "Cambridge");

        List<Transaction> transactions = Arrays.asList(
            new Transaction(brian, 2011, 300),
            new Transaction(raoul, 2012, 1000),
            new Transaction(raoul, 2011, 400),
            new Transaction(mario, 2012, 710),
            new Transaction(mario, 2012, 700),
            new Transaction(alan, 2012, 950)
        );
        
        // 1. 2011년에 일어난 모든 트랜잭션을 찾아 값을 오름차순으로 정렬하시오.
        // CODE
        
        // 2. 2011년에 일어난 모든 트랜잭션을 찾아 값을 기준으로 오름차순으로 정렬하시오.
        // CODE
        
        // 3. 거래자가 근무하는 모든 도시를 중복 없이 나열하시오.
        // CODE
        
        // 4. 케임브리지에서 근무하는 모든 거래자를 찾아서 이름순으로 정렬하시오.
        // CODE
        
        // 5. 모든 거래자의 이름을 알파벳 역순으로 정렬해서 반환하시오.
        // CODE
        
        // 6. 밀라노에 거래자가 있는가?
        // CODE
        
        // 7. 케임브리지에 거주하는 거래자의 모든 트랜잭션 값을 출력하시오.
        // CODE
        
        // 8. 전체 트랜잭션 중 최댓값은 얼마인가?
        // CODE
   }
}

 

 

정답

1. 2011년에 일어난 모든 트랜잭션을 찾아 값을 오름차순으로 정렬하시오.

List<Integer> answer1 = transactions.stream()
    .filter(t -> t.getYear() == 2011)
    .map(Transaction::getValue)
    .sorted()
    .collect(Collectors.toList());

System.out.println(answer1);

2. 2011년에 일어난 모든 트랜잭션을 찾아 값을 기준으로 오름차순으로 정렬하시오.

List<Transaction> answer2 = transactions.stream()
    .filter(t -> t.getYear() == 2011)
    .sorted(Comparator.comparing(Transaction::getValue))
    .collect(Collectors.toList());

System.out.println(answer2);

3. 거래자가 근무하는 모든 도시를 중복 없이 나열하시오.

List<String> answer3 = transactions.stream()
    .map(Transaction::getTrader)
    .map(Trader::getCITY)
    .distinct()
    .collect(Collectors.toList());

System.out.println(answer3);

4. 케임브리지에서 근무하는 모든 거래자를 찾아서 이름순으로 정렬하시오.

List<Trader> answer4 = transactions.stream()
    .map(Transaction::getTrader)
    .filter(trader -> trader.getCITY().equals("Cambridge"))
    .sorted(Comparator.comparing(Trader::getName))
    .collect(Collectors.toList());

System.out.println(answer4);

5. 모든 거래자의 이름을 알파벳 역순으로 정렬해서 반환하시오.

List<String> answer5 = transactions.stream()
    .map(Transaction::getTrader)
    .map(Trader::getName)
    .distinct()
    .sorted(Comparator.reverseOrder())
    .collect(Collectors.toList());

System.out.println(answer5);

6. 밀라노에 거래자가 있는가?

boolean answer6 = transactions.stream()
    .anyMatch(t -> t.getTrader().getCITY().equals("Milan"));

System.out.println(answer6);

7. 케임브리지에 거주하는 거래자의 모든 트랜잭션 값을 출력하시오.

List<Integer> answer7 = transactions.stream()
    .filter(t -> t.getTrader().getCITY().equals("Cambridge"))
    .map(Transaction::getValue)
    .collect(Collectors.toList());

System.out.println(answer7);

8. 전체 트랜잭션 중 최댓값은 얼마인가?

Integer answer8 = transactions.stream()
    .map(Transaction::getValue)
    .reduce(0, Integer::max);

System.out.println(answer8);