17140번 이차원 배열과 연산

https://www.acmicpc.net/problem/17140

사실 처음에 틀릴 줄 알았는데, 맞았다.

시간에 따른 Map 배열을 출력하면서 

어디가 잘못 출력되는지, 어디를 실수했는지 등등 생각하면서

구현하였다.

그리고 코드가 좀 더러워 보이는건 기분 탓일 수도 있다.

#include <iostream>
#include <queue>
#include <stack>
#include <cstdio>
#include <vector>
#include <cstring>
#include <string>
#include <math.h>
#include <algorithm>
#include <map>
#include <set>
#include <sstream>
#include <tuple>
 
#pragma warning(disable:4996)  
#pragma comment(linker, "/STACK:336777216")
 
using namespace std;
 
int Map[110][110] = { 0, }, Count[110] = { 0, };
 
int r = 0, c = 0, k = 0, mr = 0, mc = 0, time = 0;
 
bool cmp(const pair<int, int> & a, const pair<int, int> & b)
{
    if (a.second == b.second) return a.first < b.first;
    else return a.second < b.second;
}
 
int simulation()
{
    for (int t = 0; t <= 100; t++)
    {        
        if (Map[r][c] == k) return t;
 
        vector<pair<int, int> > v[110];
 
        bool use[110] = { false, };
 
        int MAX = 0;
 
        if (mr >= mc) // 행의 개수가 더 많을 때
        {
            for (int i = 0; i < mr; i++)
            {        
                memset(use, false, sizeof(use));
                memset(Count, 0, sizeof(Count));
 
                for (int j = 0; j < mc; j++)
                {
                    if (Map[i][j] == 0) continue;
 
                    ++Count[Map[i][j]];
                }
                
                for (int j = 0; j < mc; j++)
                {
                    if (Map[i][j] == 0 || use[Map[i][j]]) continue;
 
                    use[Map[i][j]] = true;
                    v[i].push_back({ Map[i][j], Count[Map[i][j]] });
                }
            
                if (v[i].size() >= 2)
                    sort(v[i].begin(), v[i].end(), cmp);
                
                int vs = v[i].size();
 
                MAX = max(MAX, 2 * vs); // 최대 열의 개수
            }
 
            memset(Map, 0, sizeof(Map)); 
            // [1, 1, 2, 2]에서 [1, 2] 이런 식으로 더 짧아 질 수 있으므로
            // 미리 0으로 초기화
 
            for (int i = 0; i < mr; i++)
                for (int j = 0, idx = 0; j < v[i].size(); j++, idx += 2)
                {
                    Map[i][idx] = v[i].at(j).first; 
                    Map[i][idx + 1] = v[i].at(j).second;
                }
 
            if (MAX >= 100) MAX = 100;
 
            mc = MAX; // 열 개수 갱신
        }
        else
        {
            for (int j = 0; j < mc; j++)
            {
                memset(use, false, sizeof(use));
                memset(Count, 0, sizeof(Count));
 
                for (int i = 0; i < mr; i++)
                {
                    if (Map[i][j] == 0) continue;
 
                    ++Count[Map[i][j]];
                }
                
                for (int i = 0; i < mr; i++)
                {
                    if (Map[i][j] == 0 || use[Map[i][j]]) continue;
 
                    use[Map[i][j]] = true;
                    v[j].push_back({ Map[i][j], Count[Map[i][j]] });
                }
 
                if (v[j].size() >= 2) sort(v[j].begin(), v[j].end(), cmp);
 
                int vs = v[j].size();
 
                MAX = max(MAX, 2 * vs); // 최대 행의 개수
            }
            
            memset(Map, 0, sizeof(Map));
 
            for (int j = 0; j < mc; j++)
                for (int i = 0, idx = 0; i < v[j].size(); i++, idx += 2)
                {
                    Map[idx][j] = v[j].at(i).first; 
                    Map[idx + 1][j] = v[j].at(i).second;
                }
 
            if (MAX >= 100) MAX = 100;
 
            mr = MAX; // 행 개수 갱신
        }
    }
 
    return -1;
}
 
int main(void)
{
    mr = mc = 3;
 
    scanf("%d %d %d", &r, &c, &k);
 
    --r; --c;
 
    for (int i = 0; i < 3; i++)
        for (int j = 0; j < 3; j++)    
            scanf("%d", &Map[i][j]);
 
    printf("%d\n", simulation());
 
    return 0;
}