기술을 기술하다

Algorithm 문제 Java Code Snippet

Collection, 람다 그리고 stream을 중심으로

Word count: 318Reading time: 1 min
2019/05/01 Share

알고리즘 문제 풀이 시 자주 사용 되는 Code Snippet 정리

한동안 풀었던 알고리즘 문제들의 소스 코드를 스스로 리뷰해보는 시간을 가졌다.
알고리즘 문제를 풀 때면 비슷한 코드 조각들을 작성하게 되는데, 그때그때 찾아보거나 생각해서 작성하려니 시간 소요가 생각보다 커지게 되었다. 이와 같은 이유로 이제껏 작성한 코드에서 자주 사용되는 조각들을 추출하여 정리해보았다.

정리해보니 자료구조의 사용과 관련된 코드들이 많았다. 앞으로 Collection과 람다를 깊게 공부하여 더 좋은 방법이 있으면 바로바로 업데이트할 생각이다.

HashMap 사용

1
2
results.putIfAbsent(rank, 0);
results.computeIfPresent(rank, (k, v) -> ++v);

List 사용

1
allCases.removeIf(i -> !Arrays.equals(result, simulate(i, ball)));

우선순위 큐 사용

1
2
3
4
public int getMinimumAverageProcessingTime(int[][] jobs) {
Queue<Job> scheduleQueue = new PriorityQueue<>(Comparator.comparing(j -> j.requestedTime));
Arrays.stream(jobs).forEach(j -> scheduleQueue.add(new Job(j[0], j[1])));
}
1
2
Queue<Integer> minHeap = new PriorityQueue<>(Comparator.naturalOrder());
Queue<Integer> maxHeap = new PriorityQueue<>(Comparator.reverseOrder());

Array를 List로

1
2
3
4
5
6
7
List<String> participantList = Arrays.asList(participant);
List<Integer> lostStudents = Arrays.stream(lost)
.boxed()
.collect(Collectors.toList());
List<Integer> lostStudents = IntStream.of(lost)
.boxed()
.collect(Collectors.toList());
1
2
3
4
List<Integer> peopleList = Arrays.stream(people)
.boxed()
.sorted(Comparator.reverseOrder())
.collect(Collectors.toList());

List를 Array로

1
2
3
return numbers.stream()
.mapToInt(i -> i)
.toArray();

Array를 Queue로

1
2
3
Queue<Integer> dateQueue = Arrays.stream(dates)
.boxed()
.collect(Collectors.toCollection(LinkedList::new));

Array 정렬

1
2
Arrays.sort(cubes, Collections.reverseOrder());
Arrays.sort(costs, Comparator.comparing(cost -> cost[2]));
1
2
3
4
5
6
7
8
9
10
int[] sorted = IntStream.of(a)
.boxed()
.sorted(Comparator.reverseOrder())
.mapToInt(i -> i)
.toArray();
int[] sortedNumbers = Arrays.stream(numbers)
.boxed()
.sorted((Integer o1, Integer o2) -> Integer.valueOf(o2 + "" + o1) - Integer.valueOf(o1 + "" + o2))
.mapToInt(i -> i)
.toArray();

List에서 최대/최소값 찾기

1
2
3
4
Integer max = listOfIntegers
.stream()
.mapToInt(v -> v)
.max().orElseThrow(NoSuchElementException::new);
1
2
3
4
Person minByAge = people
.stream()
.min(Comparator.comparing(Person::getAge))
.orElseThrow(NoSuchElementException::new);
CATALOG
  1. 1. 알고리즘 문제 풀이 시 자주 사용 되는 Code Snippet 정리
  2. 2. HashMap 사용
  3. 3. List 사용
  4. 4. 우선순위 큐 사용
  5. 5. Array를 List로
  6. 6. List를 Array로
  7. 7. Array를 Queue로
  8. 8. Array 정렬
  9. 9. List에서 최대/최소값 찾기