[프로그래머스] 순위 검색 2021 카카오 블라인드 테스트 level2

문제 : 

programmers.co.kr/learn/courses/30/lessons/72412

코딩테스트 연습 - 순위 검색

분석 :

이 문제는 문자열과 관련된 문제였다. 

(카카오에선 항상 1문제씩 문자열 관련된 문제를 내는것 같다)

 

문제에서 주어진 info와 query 배열은 1차원 배열로 되어 있어서 2차원 배열로 token 단위 마다 저장할 수 있도록 하였다. 그리고 마지막 점수에 대해선 int형 vector에 각각 받았다.

효율성 측면에선 0점을 받았으니 이렇게 푼 사람도 있구나 .. 하고 넘어가도 좋다 하하

 

sstream으로 공백 기준으로 사람을 기준으론 set배열에 담았고 query에 대해선 vector에 담았다. 

그렇게 담은 뒤에 for문을 3개나 돈다 

첫번째 for문은 query의 수만큼, 두번째 for문은 한 query당 검사할 사람을 대응시켰고 세번째 for문은 검사할 사람당 query의 토큰을 대응시켜 세번째 for문의 token을 모두 통과한 사람에 대해 cnt값을 1 증가시켜 j사람만큼 총 6 사람 검사 시킨 뒤의 cnt 값을 answer vector에 담았다. 

 

처음에 고생했던게 for문의 뜻을 짐작하는 것이었고 두번째 고생했던 부분은 isCondition = false 밑에 else isCondition = true를 달아 잘못된 값을 찾으면 바로 리턴하지 않고 다음 조건문에서 true가 되어 다음 token 검사하고 나서 true를 만든다는 점에서 애먹었다 ☹️ 

 

시간이 오래 걸렸고 좋은 코드도 아닌데 앞으로 더 많은 문제를 풀어 봐야겠다 

 

 

c++ 코드 :

#include <stdio.h>
#include <iostream>
#include <vector>
#include <sstream>
#include <set>

using namespace std;

set<string> memberSet[50001];
vector<int> memberScore; // 점수 보관용

vector<string> searchQuery[50001];
vector<int> searchScore;

vector<int> solution(vector<string> info, vector<string> query) {
     vector<int> answer;
     for(int i = 0; i< info.size(); i++){
        string s = info[i];
        stringstream ss(s);
        string str;
        int j =0;
        while(ss>>str){
            j++;
            if(j == 5){
                memberScore.push_back(stoi(str));
                continue;
            }
            memberSet[i].insert(str);
        }
    }
    
    for(int i = 0; i < query.size(); i++){
        string s = query[i];
        stringstream ss(s);
        string str;
        int j = 0;
        while(ss>>str){
            j++;
            if(str == "and"){
                continue;
            }
            if(j == 8){
                searchScore.push_back(stoi(str));
                continue;
            }
            
            searchQuery[i].push_back(str);
        }
    }
    
     for(int i = 0; i < (int)query.size(); i++){ // query index 6
        int cnt = 0;
        
        for(int j = 0; j < (int)info.size(); j++){ // person index 6
            bool isCondition = true;
            for(int k = 0; k < (int)memberSet[j].size() + 1; k++){
                if(k == 4){
                    if(memberScore.at(j) < searchScore.at(i)){
                        isCondition = false;
                        break;
                    }
                }else { //k<=3
                    string queryWord = searchQuery[i][k];
                    set<string>:: iterator iter;
                    if(queryWord == "-"){
                        continue;
                    }else{
                        iter = memberSet[j].find(queryWord);
                        if(iter == memberSet[j].end()){
                            isCondition = false;
                        }
                    }
                }
            }
            
            if(isCondition == true){
                cnt++;
            }
        }
        answer.push_back(cnt);
    }
    return answer;
}

 

결과 :