BOJ 13144 List of Unique Numbers

https://www.acmicpc.net/problem/13144

문제

길이가 N인 수열이 주어질 때, 수열에서 연속한 1개 이상의 수를 뽑았을 때 같은 수가 여러 번 등장하지 않는 경우의 수를 구하는 프로그램을 작성하여라.

---

입력

첫 번째 줄에는 수열의 길이 N이 주어진다. (1 ≤ N ≤ 100,000)

두 번째 줄에는 수열을 나타내는 N개의 정수가 주어진다. 수열에 나타나는 수는 모두 1 이상 100,000 이하이다.

---

출력

조건을 만족하는 경우의 수를 출력한다.

---

예제 입력 1

5
1 2 3 4 5

예제 출력 1

15

예제 입력 2

5
1 2 3 1 2

예제 출력 2

12

… 이하 예제 생략

---

코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.HashSet;
import java.util.Set;
import java.util.StringTokenizer;

public class Main {
    static int N;
    static int[] sequence;

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        N = Integer.parseInt(br.readLine());
        sequence = new int[N];
        StringTokenizer st = new StringTokenizer(br.readLine());
        for (int i = 0; i < N; i++) {
            sequence[i] = Integer.parseInt(st.nextToken());
        }

        Set<Integer> selected = new HashSet<>();
        long answer = 0;
        int leftIdx = 0;
        int rightIdx = 0;
        while (rightIdx < N) {
            if(!selected.contains(sequence[rightIdx])) {
                selected.add(sequence[rightIdx]);
                rightIdx++;
                answer += rightIdx - leftIdx;
            } else {
                selected.remove(sequence[leftIdx]);
                leftIdx++;
            }
        }

        System.out.println(answer);
    }
}