Magnet

package IM; import java.io.File; import java.io.FileNotFoundException; import java.util.Scanner; public class Magnet { static int T, N, M, D, x, y, Ans; static int[][] map; static int[] arrDraw; static final int UP = 1, RIGHT = 2, DOWN = 3, LEFT = 4; public static int checkRotate(int d){ if(arrDraw[d] == 1){ return UP; }else if(arrDraw[d] == 2){ return RIGHT; }else if(arrDraw[d] == 3){ return DOWN; }else return LEFT; } public static int magnet(int x, int y, int rotate){ int moves = 0; for(int k = 1; x+k < N; k++){ if(rotate == DOWN && map[x+k][y] == 0){ moves++; }else break; } for(int k = 1; x-k>=0; k++){ if(rotate == UP && map[x-k][y] == 0){ moves++; }else break; } return moves; } public static void main(String[] args) throws FileNotFoundException { Scanner sc = new Scanner(new File("input_magnet.txt")); T = sc.nextInt(); for(int tc = 1; tc <= 1; tc++){ N = sc.nextInt(); M = sc.nextInt(); D = sc.nextInt(); map = new int[N][M]; arrDraw = new int[D]; Ans = 0; // Doc data vao mang Draw for(int i = 0; i < D; i++){ arrDraw[i] = sc.nextInt(); } // Doc data vao mang MN for(int i = 0; i < N; i++){ for(int j = 0; j < M; j++){ map[i][j] = sc.nextInt(); System.out.print(map[i][j]+" "); if(map[i][j] == 2){ x = i; y = j; } } System.out.println(); } // Xu ly for(int d = 0; d < D; d++){ Ans += magnet(x,y,checkRotate(arrDraw[d])); } //In ket qua System.out.println("Case#"+tc+" "+Ans); } } }