프로그래머스 [Kakao] 신고 결과 받기 (Java)

링크

문제 링크

문제 설명

신입사원 무지는 게시판 불량 이용자를 신고하고 처리 결과를 메일로 발송하는 시스템을 개발하려 합니다. 무지가 개발하려는 시스템은 다음과 같습니다.

  • 각 유저는 한 번에 한 명의 유저를 신고할 수 있습니다.
    • 신고 횟수에 제한은 없습니다. 서로 다른 유저를 계속해서 신고할 수 있습니다.
    • 한 유저를 여러 번 신고할 수도 있지만, 동일한 유저에 대한 신고 횟수는 1회로 처리됩니다.
  • k번 이상 신고된 유저는 게시판 이용이 정지되며, 해당 유저를 신고한 모든 유저에게 정지 사실을 메일로 발송합니다.
    • 유저가 신고한 모든 내용을 취합하여 마지막에 한꺼번에 게시판 이용 정지를 시키면서 정지 메일을 발송합니다.

다음은 전체 유저 목록이 ["muzi", "frodo", "apeach", "neo"]이고, k = 2(즉, 2번 이상 신고당하면 이용 정지)인 경우의 예시입니다.

각 유저별로 신고당한 횟수는 다음과 같습니다.

위 예시에서는 2번 이상 신고당한 "frodo"와 "neo"의 게시판 이용이 정지됩니다. 이때, 각 유저별로 신고한 아이디와 정지된 아이디를 정리하면 다음과 같습니다.

따라서 "muzi"는 처리 결과 메일을 2회, "frodo"와 "apeach"는 각각 처리 결과 메일을 1회 받게 됩니다.

이용자의 ID가 담긴 문자열 배열 id_list, 각 이용자가 신고한 이용자의 ID 정보가 담긴 문자열 배열 report, 정지 기준이 되는 신고 횟수 k가 매개변수로 주어질 때, 각 유저별로 처리 결과 메일을 받은 횟수를 배열에 담아 return 하도록 solution 함수를 완성해주세요.

제한사항

  • 2 ≤ id_list의 길이 ≤ 1,000
    • 1 ≤ id_list의 원소 길이 ≤ 10
    • id_list의 원소는 이용자의 id를 나타내는 문자열이며 알파벳 소문자로만 이루어져 있습니다.
    • id_list에는 같은 아이디가 중복해서 들어있지 않습니다.
  • 1 ≤ report의 길이 ≤ 200,000
    • 3 ≤ report의 원소 길이 ≤ 21
    • report의 원소는 "이용자id 신고한id"형태의 문자열입니다.
    • 예를 들어 "muzi frodo"의 경우 "muzi"가 "frodo"를 신고했다는 의미입니다.
    • id는 알파벳 소문자로만 이루어져 있습니다.
    • 이용자id와 신고한id는 공백(스페이스)하나로 구분되어 있습니다.
    • 자기 자신을 신고하는 경우는 없습니다.
  • 1 ≤ k ≤ 200, k는 자연수입니다.
  • return 하는 배열은 id_list에 담긴 id 순서대로 각 유저가 받은 결과 메일 수를 담으면 됩니다.

입출력 예

입출력 예 설명

입출력 예 #1

문제의 예시와 같습니다.

입출력 예 #2

"ryan"이 "con"을 4번 신고했으나, 주어진 조건에 따라 한 유저가 같은 유저를 여러 번 신고한 경우는 신고 횟수 1회로 처리합니다. 따라서 "con"은 1회 신고당했습니다. 3번 이상 신고당한 이용자는 없으며, "con"과 "ryan"은 결과 메일을 받지 않습니다. 따라서 [0, 0]을 return 합니다.

풀이

  1. 신고 당한 횟수를 담은 <String, Integer> 해시맵과 <String, ArrayList>로 자신을 신고한 사람의 리스트를 저장하는 해시맵 두 개를 선언하자!
  2. 첫 번째 for문에서 누가 몇 신고당했는지, 그 사람을 누가 신고했는지에 대해 put 한다.
  3. 두 번째 for문에서는 k회 이상 신고당한 사람이 있다면 id_list 속 그 사람을 신고한 사람의 index를 받아서 ++ 해준다.

이 문제는 저어어엉말 오래걸렸다... ㅠㅠ

처음에 신고 당한 횟수만 담은 해시맵 하나만 써서 두 번째 포문에서 report배열만큼 계속 훑으니 정답은 맞는데 두 가지 경우에서 시간 초과가 떠버렸다.

그래서 자료구조를 하나 더 선언하는데(Key와 Value)중 Key는 당연히 신고당한 사람의 이름이지만, Value를 어떻게 선언해야 깔끔할까에 대한 고민을 오랫동안 했다.

다양한 시도를 해보았다.

  • <String, String>으로 신고를 한 사람의 이름을 공백을 기준으로 이어붙이자
  • List<List< String>>

첫 번째, 신고했던 사람의 이름을 공백을 두고 이어붙여서 나중에 길다란 String속에서 id_list 속의 사람을 대조하는 방법인데 왜인진 모르겠는데 자꾸 실패했다. (아직도 이유를 모름..) 아 근데 만약 eee라는 이름을 가진 사람이 eeee라는 이름을 가진 사람에게 동일 인물로 걸렸던 것 같은데 이 부분 때문이라기엔 정답률이 너무 최악이었다..

그래서 두 번째 리스트 속의 리스트로 관리를 하려했는데 이것 역시 실패... 이건 문법적인 이유때문인 것 같았다. 인터페이스와 구현체간의 호환? 때문인듯?

그래서 두 번째를 조금 더 발전시킨것이 Map<String, ArrayList>다. get으로 받은 뒤에 ArrayList로 처리하니 바로 성공!

(근데 다른사람의 풀이를 보고 솔직히 벽느꼈다..)

2022-01-20 수정 // 아 근데 이 문제 생각해보니 mail_to 해시맵의 Value값인 ArrayList의 Size가 곧 몇번 신고당했는지 담은거니까 훨씬 더 스마트하게 풀 수 있었네ㅠㅠ

코드

import java.util.*;
class Solution {
  public int index_of_list(String[] id_list, String s){
      int i;
      for(i = 0; i < id_list.length; i++)
          if(id_list[i].equals(s))
              return i;
      return i;
  }
  public int[] solution(String[] id_list, String[] report, int k) {
      int[] answer = new int[id_list.length];
      int i, j;
      Map<String, Integer> num_reported = new HashMap<>();
      Map<String, ArrayList<String>> mail_to = new HashMap<>();
      ArrayList<String> comp;
      for(i = 0; i < report.length; i++) {
          String[] info = report[i].split(" ");
          if(mail_to.get(info[1]) == null) {
              mail_to.put(info[1], new ArrayList(Arrays.asList(info[0])));
              num_reported.put(info[1], 1);
          }
          else {
              if(mail_to.get(info[1]).indexOf(info[0]) < 0){
                  num_reported.put(info[1], num_reported.get(info[1]) + 1);
                  mail_to.get(info[1]).add(info[0]);           
              }                
          }
      }
      for(i = 0; i < id_list.length; i++) {
          if(num_reported.getOrDefault(id_list[i], 0) >= k) {
              comp = mail_to.get(id_list[i]);
              for(j = 0; j < comp.size(); j++)
              {
                  answer[index_of_list(id_list,comp.get(j))]++;
              }
          }
      }
      return answer;
  }
}

좋은 웹페이지 즐겨찾기