[프로그래밍] 프로그래머스 예제문제 풀어보기 <신고 결과 받기> 재도전(66.7점->75점)

프로그래머스 예제문제를 풀다보면 시간이 참 빨리 지나간다. 작성한 코드는 몇 줄 안되는데 너무 느린것 같다.
주인 닮아서 코드도 느린지 시간초과로 실패하였다.
코드의 작동시간을 느리게 만드는 원인이 무엇일까...

수정 아이디어))
알고리즘 자체를 바보같이 함
record를 HashMap으로 만듦(신고자=피신고자)
for 문 id별 반복
id 키값의 value를 카운팅하고 k이상일 경우
mailCount ++
answer에 바인딩

이것도 뭔가 아님..

이따 얼른 해봐야겠다

import java.util.ArrayList;
import java.util.HashMap;
import java.util.Map;
 
class Solution {
    public int[] solution(String[] id_list, String[] report, int k) {
 
        
        ArrayList<String>successReporterList = new ArrayList<>();
        int[] answer = new int[id_list.length];
        
        for(String member: id_list) {
            Map<String,String> thisMemberReportRecord = new HashMap<>();
            
            for(String record:report) {
                String repoter= record.split(" ")[0];
                String defendant = record.split(" ")[1];
                
                if(member.equals(defendant)) {
                    thisMemberReportRecord.put(repoter, member);    
                }
            }
            if(thisMemberReportRecord.size()>=k) {
                String[] thisMemberReporter = thisMemberReportRecord.keySet().toArray(new String[0]);
                for(String TMR:thisMemberReporter) {                    
                    successReporterList.add(TMR);                                        
                }
            }
        }
        
        for(int j = 0; j<id_list.length; j++) {
            int successCount = 0;
            for(String successMan: successReporterList) {
                if (id_list[j].equals(successMan)){
                    successCount++;
                }    
            }
            answer[j] = successCount;
        }
    
        for(int h=0; h<id_list.length;h++) {
        System.out.println(answer[h]);
        }        
                
                
        return answer;
    }
}

 

출처: 프로그래머스 채점결과, 처음 시도보다는 나아짐.

import java.util.ArrayList;
import java.util.HashMap;
import java.util.Map;
import java.util.Set;
import java.util.HashSet;
 
class Solution {
    public int[] solution(String[] id_list, String[] report, int k) {
 
        
        ArrayList<String>successReporterList = new ArrayList<>();
        int[] answer = new int[id_list.length];
        
        for(String member: id_list) {
            Set<String> thisMemberReportRecord = new HashSet<>();
            
            for(String record:report) {
                String repoter= record.split(" ")[0];
                String defendant = record.split(" ")[1];
                
                if(member.equals(defendant)) {
                    thisMemberReportRecord.add(repoter);    
                }
            }
            if(thisMemberReportRecord.size()>=k) {
                String[] thisMemberReporter = thisMemberReportRecord.toArray(new String[0]);
                for(String TMR:thisMemberReporter) {                    
                    successReporterList.add(TMR);                                        
                }
            }
        }
        
        for(int j = 0; j<id_list.length; j++) {
            int successCount = 0;
            for(String successMan: successReporterList) {
                if (id_list[j].equals(successMan)){
                    successCount++;
                }    
            }
            answer[j] = successCount;
        }
    
        for(int h=0; h<id_list.length;h++) {
        System.out.println(answer[h]);
        }        
                
                
        return answer;
    }
}

 

출처:프로그래머스 채점결과

HashMap으로 불필요하게 value 값을 넣었던 것을 HashSet로 변경하였더니 점수가 소폭 올라감.