[알고리즘]Sorting & Searching - 중복 확인

5. 중복 확인
N명의 학생들이 적어낸 숫자 중 중복된 숫자가 존재하면 D(duplication)를 출력하고,
N명이 모두 각자 다른 숫자를 적어냈다면 U(unique)를 출력하는 프로그램

입력) 첫 번째 줄에 자연수 N(5<=N<=100,000)이 주어집니다.
두 번째 줄에 학생들이 적어 낸 N개의 자연수가 입력됩니다.
출력) 첫 번째 줄에 D 또는 U를 출력합니다.

 

 

1)

import java.util.Arrays;
import java.util.Scanner;

public class Main {
	
	public static String solution(int n, int[] arr) {
		String answer = "U"; // Unique
		
		Arrays.sort(arr); // 배열 오름차순 정렬
		for (int i=0; i<n-1; i++) { // 이웃한 두 원소 비교
			if (arr[i]==arr[i+1]) return "D";
		}
		return answer;
	}

	public static void main(String[] args) {
		Scanner scanner = new Scanner(System.in);
		int n = scanner.nextInt();
		int[] arr = new int[n];
		for (int i=0; i<n; i++) {
			arr[i] = scanner.nextInt();
		}
		System.out.println(solution(n,arr));
	}

}

 

2) 그냥 TreeSet, HashMap 등 사용해도 됨

import java.util.Scanner;
import java.util.TreeSet;

public class Main {
	
	public static char solution(int n, int[] arr) {
		char answer = 'U'; // Unique
		TreeSet<Integer> tree = new TreeSet<>();
		
		for (int x : arr) {
			tree.add(x);		
		}
		if (tree.size()!=n) return 'D'; // Duplication
		return answer;
	}

	public static void main(String[] args) {
		Scanner scanner = new Scanner(System.in);
		int n = scanner.nextInt();
		int[] arr = new int[n];
		for (int i=0; i<n; i++) {
			arr[i] = scanner.nextInt();
		}
		System.out.println(solution(n,arr));
	}

}

---

결과

입력
8
20 25 52 30 39 33 43 33

출력
D