[프로그래머스] 단어 퍼즐

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

풀이

1. t의 길이 + 1 만큼 dp를 생성하고 INT_MAX로 초기화
2. 입력받은 문자열 t를 반복문 i을 돌면서 t.length()-1부터 0까지 탐색
3. 2의 반복문 안에 새로운 반복문 j로 i부터 t.length()까지 탐색하며 단어 조각을 생성
4. 단어 조각이 기존에 존재하는 단어 조각인지 확인
5. 존재하는 단어 조각이면 dp[i] = min(dp[i], dp[j+1])으로 값 갱신
6. 반복문이 완료되면 dp[0] 값을 확인

코드

#include <iostream>
#include <string>
#include <vector>
#include <climits>
#include <set>
using namespace std;

int solution(vector<string> strs, string t)
{
    int answer = 0;
    set<string> s;
    vector<int> dp(t.length()+1, INT_MAX);

    // set에 단어 조각 삽입
    for(int i=0; i<strs.size(); i++)
        s.insert(strs[i]);

    dp[t.length()] = 0;

    // t의 뒤부터 탐색
    for(int i=t.length()-1; i>=0; i--){
        string tmp = "";
        // i, j 값을 이용해서 단어 조각 생성
        for(int j=i; j<t.length(); j++){
            tmp += t[j];
            if(i + 5 < j) break;            
            // set에 있는 단어조각인지 검사
            if(s.find(tmp) != s.end()){
                if(dp[j+1] != INT_MAX){
                    dp[i] = min(dp[i], dp[j+1] + 1);
                }
            }
        }
    }
    answer = dp[0] != INT_MAX ? dp[0] : -1;
    return answer;
}