백준 17144번 미세먼지 안녕! ( C++ )

문제

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

 

해설

어렵진 않았으나 몹시 귀찮았던 문제

잘 읽고 예외처리만 잘 하면 문제 없이 풀 수 있음

 

코드

#include <iostream>
#include <algorithm>
#include <vector>
 
using namespace std;
 
int mp1[54][54], mp2[54][54], ret, r, c, t;
vector<int> robot;
int dy[4] = {-1, 0, 1, 0};
int dx[4] = {0, 1, 0, -1};
 
void go(int a1[54][54], int a2[54][54])
{
  for (int i = 0; i < r; i++)
  {
    for (int j = 0; j < c; j++)
    {
      a2[i][j] += a1[i][j];
      if (a1[i][j] > 0)
      {
        int mun = a1[i][j] / 5;
        for (int k = 0; k < 4; k++)
        {
          int ny = i + dy[k];
          int nx = j + dx[k];
          if (ny >= 0 && ny < r && nx >= 0 && nx < c && a1[ny][nx] >= 0)
          {
            a2[ny][nx] += mun;
            a2[i][j] -= mun;
          }
        }
      }
    }
  }
}
 
void clear_air(int a[54][54])
{
  int y1 = robot[0];
  int y2 = robot[1];
 
  for (int i = y1 - 1; i > 0; i--)
  {
    a[i][0] = a[i - 1][0];
  }
  for (int i = 0; i < c - 1; i++)
  {
    a[0][i] = a[0][i + 1];
  }
  for (int i = 0; i < y1; i++)
  {
    a[i][c - 1] = a[i + 1][c - 1];
  }
  for (int i = c - 1; i > 1; i--)
  {
    a[y1][i] = a[y1][i - 1];
  }
 
  for (int i = y2 + 1; i < r - 1; i++)
  {
    a[i][0] = a[i + 1][0];
  }
  for (int i = 0; i < c - 1; i++)
  {
    a[r - 1][i] = a[r - 1][i + 1];
  }
  for (int i = r - 1; i > y2; i--)
  {
    a[i][c - 1] = a[i - 1][c - 1];
  }
  for (int i = c - 1; i > 1; i--)
  {
    a[y2][i] = a[y2][i - 1];
  }
  a[y1][1] = 0;
  a[y2][1] = 0;
}
 
int main()
{
  cin >> r >> c >> t;
 
  for (int i = 0; i < r; i++)
  {
    for (int j = 0; j < c; j++)
    {
      cin >> mp1[i][j];
      if (mp1[i][j] == -1)
      {
        robot.push_back(i);
      }
    }
  }
 
  bool flag = true;
  while (t)
  {
    t--;
    if (flag)
    {
      go(mp1, mp2);
      clear_air(mp2);
      for (int i = 0; i < r; i++)
      {
        for (int j = 0; j < c; j++)
        {
          mp1[i][j] = 0;
        }
      }
    }
    else
    {
      go(mp2, mp1);
      clear_air(mp1);
      for (int i = 0; i < r; i++)
      {
        for (int j = 0; j < c; j++)
        {
          mp2[i][j] = 0;
        }
      }
    }
    flag = !flag;
  }
 
  ret = 0;
  if (flag)
  {
    for (int i = 0; i < r; i++)
    {
      for (int j = 0; j < c; j++)
      {
        ret += mp1[i][j];
      }
    }
  }
  else
  {
    for (int i = 0; i < r; i++)
    {
      for (int j = 0; j < c; j++)
      {
        ret += mp2[i][j];
      }
    }
  }
 
  ret += 2;
 
  cout << ret;
  return 0;
}