BOJ1343 : 폴리오미노 (Silver 5)

#include <iostream>
#include <vector>
#include <string.h>
using namespace std;

string poliomino[501];
void polifunc()
{
    poliomino[0] = ".";
    poliomino[1] = "-1";
    poliomino[2] = "BB";
    poliomino[3] = "-1";
    poliomino[4] = "AAAA";
    for (int i = 5; i <= 500; i++)
    {
        if (poliomino[i - 2] != "-1")
        {
            if (poliomino[i - 4] != "-1")
            {
                string tmp = poliomino[i - 4] + "AAAA";
                string tmp2 = poliomino[i - 2] + "BB";
                if (tmp > tmp2)
                {
                    poliomino[i] = tmp2;
                }
                else
                {
                    poliomino[i] = tmp;
                }
            }
            else
            {
                poliomino[i] = poliomino[i - 2] + "BB";
            }
        }

        else
            poliomino[i] = "-1";
    }
}

int main()
{
    string poli;
    cin >> poli;
    polifunc();
    string result;
    int count = 0;
    for (int i = 0; i < poli.length(); i++)
    {

        if (poli[i] == '.' && count != 0)
        {
            if (poliomino[count] == "-1")
            {
                cout << -1;
                return 0;
            }
            result += (poliomino[count]);
            count = 0;
        }
        if (poli[i] == '.' && count == 0)
            result += (poliomino[0]);
        else
            count++;
    }
    if (count)
    {
        if (poliomino[count] == "-1")
        {
            cout << -1;
            return 0;
        }
        result += (poliomino[count]);
    }
    for (int i = 0; i < result.length(); i++)
        cout << result[i];
}

막판에 예외 찾는게 조금 빡셌지만 그래도 초반에 아이디어 떠올리기는 쉬웠다.