[백준 15662] 톱니바퀴2 -JAVA //le_effort//

https://suhyeokeee.tistory.com/34

[백준 14891] 톱니바퀴 -JAVA //le_effort//

 

이전 포스팅과 똑같은 알고리즘이기 때문에 설명을 적진 않겠다.

설명이 필요하다면 위의 링크를 보길 바란다.

 

위 문제와 똑같다. 

 

다만 다른것 이라면 톱니바퀴의 개수가 4개가 아니라 t개라는 것

이것만 바꿔주면 된다.

 

import java.io.*;
import java.util.*;
public class Main {
    static int t,k;
    static int map[][];
    static boolean[] visited ;
    static int copy_map[][];
    static int sum=0;
    static Queue<Node> q = new LinkedList<>();
    static ArrayList<Node>list = new ArrayList<>();
    public static void main(String[] args) throws IOException{
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        
        t = Integer.parseInt(br.readLine());
        
        map = new int[t+1][9];
        copy_map = new int[t+1][9];
        for(int i=1; i<=t; i++) {
            String str = br.readLine();
            for(int j=1; j<=8; j++) {
                map[i][j] = str.charAt(j-1)-'0';
            }
        }
        k = Integer.parseInt(br.readLine());
        
        for(int i=0; i<k; i++) {
            String[] t = br.readLine().split(" ");
            int num = Integer.parseInt(t[0]);
            int d = Integer.parseInt(t[1]);
            q.add(new Node(num,d));
        }
        solve();
    }
    public static void solve() {
        while(!q.isEmpty()) {
            list.clear();
            visited= new boolean[t+1];
            Node a= q.poll();
            int num = a.num;
            int d = a.d;
            visited[num]=true;
            can_turn(num,d);
            for(int i=0; i<list.size(); i++) {
                turn(list.get(i).num,list.get(i).d);
            }
        }
        for(int i=1; i<=t; i++) {
            if(map[i][1]==1) {
                sum++;
            }
        }
        System.out.println(sum);
    }
    public static void turn(int num, int d) {
        for(int i=1; i<=t; i++) {
            for(int j=1; j<=8; j++) {
                copy_map[i][j]=map[i][j];
            }
        }
        if(d==1) {
            for(int i=1; i<=8; i++) {
                if(i==1) {
                    map[num][1] = copy_map[num][8];
                }
                else {
                    map[num][i] = copy_map[num][i-1];
                }
            }
        }
        else {
            for(int i=8; i>=1; i--) {
                if(i==8) {
                    map[num][8]=copy_map[num][1];
                }
                else {
                    map[num][i] = copy_map[num][i+1];
                }
            }
        }
    }
    public static void can_turn(int num, int d) {
        list.add(new Node(num,d));
        if(num-1>=1 && !visited[num-1] && map[num][7] != map[num-1][3]) {
            visited[num-1] = true;
            can_turn(num-1,d*-1);
        }
        if(num+1<=t && !visited[num+1] && map[num][3] != map[num+1][7]) {
            visited[num+1] = true;
            can_turn(num+1,d*-1);
        }
    }
}
class Node{
    int num,d;
    Node(int num, int d){
        this.num=num;
        this.d=d;
    }
}