[SWEA] 6019. 추억의 2048게임

이 문제는 복잡해보이지만 문제에 주어진 조건 그대로 하면되는 단순구현 문제이다.

문제에서 가장어려웠던 요소는 4가지 방향을 생각해주는 것과 0을 한쪽으로 옮기는 과정이었다. 0을 한쪽으로 옮기는 메소드를 moveZero~(row or col) 이런식으로 구성하였는데, 다양한 문제에서 빈번하게 사용되는 코드이므로 숙지하고 있으면 많은 도움이 될 것 같다.

package swea;
 
import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.StringTokenizer;
 
public class Solution_d4_6109_추억의2048게임 {
    
    static int[][] map;
    static int N;
    
    
    public static void main(String[] args) throws Exception {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st  = null;
        
        int T=Integer.parseInt(br.readLine());
        for(int tc=1;tc<=T;tc++) {
            st=new StringTokenizer(br.readLine());
            N=Integer.parseInt(st.nextToken());
            char c=st.nextToken().charAt(0);
            
            map=new int[N][N];
            for(int i=0;i<N;i++) {
                st=new StringTokenizer(br.readLine());
                for(int j=0;j<N;j++) {
                    map[i][j]=Integer.parseInt(st.nextToken());
                }
            }
            
            if(c=='u') {
                up();
            }else if(c=='r') {
                right();
            }else if(c=='d') {
                down();
            }else if(c=='l') {
                left();
            }
            
            System.out.println("#"+tc);
            for(int i=0;i<N;i++) {
                for(int j=0;j<N;j++) {
                    System.out.print(map[i][j]+" ");
                }
                System.out.println();
            }
            
        }
    }
    
    static void up() {
        for(int col=0;col<N;col++)
            moveZeroDown(col);
        
        for(int col=0;col<N;col++) {
            for(int row=0;row<=N-2;row++) {
                if(map[row][col]!=0 && checkPow2(map[row][col]+map[row+1][col])) {
                    map[row][col]=map[row][col]+map[row+1][col];
                    map[row+1][col]=0;
                    moveZeroDown(col);
                }
            }
        }
    }
    static void right() {
        for(int row=0;row<N;row++)
            moveZeroLeft(row);
        
        for(int row=0;row<N;row++) {
            for(int col=N-1;col>=1;col--) {
                if(map[row][col]!=0 && checkPow2(map[row][col]+map[row][col-1])) {
                    map[row][col]=map[row][col]+map[row][col-1];
                    map[row][col-1]=0;
                    moveZeroLeft(row);
                }
            }
        }
    }
    static void down() {
        for(int col=0;col<N;col++)
            moveZeroUp(col);
        
        for(int col=0;col<N;col++) {
            for(int row=N-1;row>=1;row--) {
                if(map[row][col]!=0 && checkPow2(map[row][col]+map[row-1][col])) {
                    map[row][col]=map[row][col]+map[row-1][col];
                    map[row-1][col]=0;
                    moveZeroUp(col);
                }
            }
        }
    }
    static void left() {
        for(int row=0;row<N;row++)
            moveZeroRight(row);
        
        for(int row=0;row<N;row++) {
            for(int col=0;col<=N-2;col++) {
                if(map[row][col]!=0 && checkPow2(map[row][col]+map[row][col+1])) {
                    map[row][col]=map[row][col]+map[row][col+1];
                    map[row][col+1]=0;
                    moveZeroRight(row);
                }
            }
        }
    }
    
    static void moveZeroLeft(int row) {
        for(int i=N-1;i>=1;i--) {
            if(map[row][i]==0) {
                for(int j=i-1;j>=0;j--) {
                    if(map[row][j]!=0) {
                        map[row][i]=map[row][j];
                        map[row][j]=0;
                        break;
                    }
                }
            }
        }
    }
    static void moveZeroRight(int row) {
        for(int i=0;i<=N-2;i++) {
            if(map[row][i]==0) {
                for(int j=i+1;j<=N-1;j++) {
                    if(map[row][j]!=0) {
                        map[row][i]=map[row][j];
                        map[row][j]=0;
                        break;
                    }
                }
            }
        }
    }
    
    static void moveZeroDown(int col) {
        for(int i=0;i<=N-2;i++) {
            if(map[i][col]==0) {
                for(int j=i+1;j<=N-1;j++) {
                    if(map[j][col]!=0) {
                        map[i][col]=map[j][col];
                        map[j][col]=0;
                        break;
                    }
                }
            }
        }
        
    }
    
    static void moveZeroUp(int col) {
        for(int i=N-1;i>=1;i--) {
            if(map[i][col]==0) {
                for(int j=i-1;j>=0;j--) {
                    if(map[j][col]!=0) {
                        map[i][col]=map[j][col];
                        map[j][col]=0;
                        break;
                    }
                }
            }
        }
    }
    
    static boolean checkPow2(int num) {
        int mod;
        while(num!=1) {
            mod=num%2;
            if(mod==0) {
                num=num/2;
            }
            else
                return false;
        }
        
        
        return true;
    }
    
}