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 magnet(int i, int j, int rotate) { int moves = 0; while (true) { if (rotate == UP) { i--; } else if (rotate == DOWN) { i++; } else if (rotate == RIGHT) { j++; } else if (rotate == LEFT) { j--; } if (i >= 0 && i < N && j >= 0 && j < M && map[i][j] == 0) { x = i; y = j; 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 <= T; 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, arrDraw[d]); } // In ket qua System.out.println("Case#" + tc + " " + Ans); } } }