[Baekjoon] 21149_Unread Messages

문제(출처: https://www.acmicpc.net/problem/21149)

---

< Unread Messages >

 

문제 풀이 

 

HashMap에 마지막으로 메시지를 보낸 시간을 저장한다. 현재 시간에서 마지막으로 메시지를 보낸 시간을 빼면 그 사람의 읽지 않은 메시지의 개수를 알 수 있다.

* 모든 사람의 읽지 않은 메시지의 총합은 long 범위이다.

 

my solution (Java)

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.HashMap;
import java.util.StringTokenizer;

public class _21149_ { // Unread Messages

	public static void main(String[] args) throws IOException {
		BufferedReader bf = new BufferedReader(new InputStreamReader(System.in));
		BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
		StringTokenizer st = new StringTokenizer(bf.readLine());

		int n = Integer.parseInt(st.nextToken());
		int m = Integer.parseInt(st.nextToken());

		long sum = 0;
		HashMap<Integer, Integer> map = new HashMap<>();
		for (int i = 0; i < m; i++) {
			int s = Integer.parseInt(bf.readLine());

			if (map.containsKey(s)) {
				sum -= (i - map.get(s) - 1);
			} else {
				sum -= i;
			}
			map.put(s, i);

			sum += (n - 1);

			bw.write(sum + "\n");
		}
		bw.flush();
	}
}

변수)
n, m : 그룹의 총 인원수, 전송된 메시지의 개수
sum : 모든 사람의 읽지 않은 메시지의 총합
map : HashMap <사람 번호, 마지막으로 메시지를 보낸 시간>

 

그룹의 총 인원 수와 전송된 메시지의 개수를 입력받는다. 전송된 메시지의 개수만큼 다음 과정을 반복한다.

 

각 메시지를 보낸 사람의 번호를 입력받는다. 그 사람이 HashMap에 key값으로 있다면 sum - ( 현재 시간 - 마지막으로 메시지를 보낸 시간 - 1)을 하고, key 값으로 없다면 sum -현재시간을 한 후 HashMap의 값을 업데이트한다. 메시지를 보낸 사람을 제외하고는 메시지가 하나씩 쌓였으므로 sum + (n-1)을 한 후 sum 값을 출력한다.

---

---