[프로그래머스 Lv 0] 안전지대 C++

https://school.programmers.co.kr/learn/courses/30/lessons/120866

프로그래머스

 

 

 

- 폭탄 주위에 X 표시를 2값으로 넣은 뒤, 전체에서 0의 갯수를 찾으려고 했다

#include <string>
#include <vector>
using namespace std;

int dir[8][2] = {{-1,0},{1,0} , {0,1},{0,-1},{-1,-1},{1,-1},{1,1},{-1,1}};

int solution(vector<vector<int>> board) {
    int answer = 0;

    for(int i=0;i<board.size();i++)
    {
        for(int j=0;j<board[i].size();j++)
        {
            if(1 == board[i][j])
            {
                for(int k=0;k<8;k++)
                {               
                    const int newX = i+dir[k][0];
                    const int newY = j+dir[k][1];
                    if(newX <0 || newY <0 || board.size() <= newX || board.size() <=newY)
                    {
                        continue;
                    }
                    if(1!=board[newX][j+dir[k][1]])
                    {
                       board[newX][newY] = 2;
                    }
                }
            }      
            
        }

    }
    
    for(int i=0;i<board.size();i++)
    {
        for(int j=0;j<board[i].size();j++)
        {
            if(board[i][j] == 0)
            {
                answer++;
            }
        }
    }
    return answer;
}

 

코드를 보다보니 좀 더 짧게 만들고 싶은 생각에 좀 더 찾아봤다

 

#include <string>
#include <vector>
#include <algorithm>
using namespace std;

int dir[8][2] = {{-1,0},{1,0} , {0,1},{0,-1},{-1,-1},{1,-1},{1,1},{-1,1}};

int solution(vector<vector<int>> board) {
    int answer = 0;

    for(int i=0;i<board.size();i++)
    {
        for(int j=0;j<board[i].size();j++)
        {
            if(1 == board[i][j])
            {
                for(int k=0;k<8;k++)
                {               
                    const int newX = i+dir[k][0];
                    const int newY = j+dir[k][1];
                    if(newX <0 || newY <0 || board.size() <= newX || board.size() <=newY)
                    {
                        continue;
                    }
                    if(1!=board[newX][j+dir[k][1]])
                    {
                       board[newX][newY] = 2;
                    }
                }
            }      
            
        }

    }
    
    //수정!
    for(const vector<int>& row : board)
    {
        answer += std::count(row.begin(),row.end(),0);
    }
    return answer;
}

 

std::count 함수를 이용해 0 카운트 순환부분을 줄였다. 2차원 배열에서 특정 값의 갯수를 셀때 수정한 방식으로 하면 좀 더 괜찮다고 생각한다.