題目描述
假定街道是棋盤型的，每格距離相等，車輛通過每格街道需要時間均為 timePerRoad；
街道的街口（交叉點）有交通燈，燈的周期 T(=lights[row][col]) 各不相同；
車輛可直行、左轉和右轉，其中直行和左轉需要等相應 T 時間的交通燈才可通行，右轉無需等待。
現給出 n*m 個街口的交通燈周期，以及起止街口的坐標，計算車輛經過兩個街口的最短時間。
其中：
1）起點和終點的交通燈不計入時間，且可以任意方向經過街口 不可超出 n*m 個街口，不可跳躍，但邊線也是道路（即 lights[0][0] -> lights[0][1] 是有效路徑）
入口函數定義:
/**
* lights : n*m 個街口每個交通燈的周期，值范圍[0,120]，n 和 m 的范圍為[1,9]
* timePerRoad : 相鄰兩個街口之間街道的通過時間,范圍為[0,600]
* rowStart : 起點的行號
* colStart : 起點的列號
* rowEnd : 終點的行號
* colEnd : 終點的列號
* return : lights[rowStart][colStart] 與 lights[rowEnd][colEnd] 兩個街口之間的最短通行時間
*/
int calcTime(int[][] lights,int timePerRoad,int rowStart,int colStart,int rowEnd,int colEnd)示例一
輸入
[[1,2,3],[4,5,6],[7,8,9]],60,0,0,2,2輸出
245說明
行走路線為 (0,0) -> (0,1) -> (1,1) -> (1,2) -> (2,2) 走了 4 格路，2 個右轉，1 個左轉，共耗時 60+0+60+5+60+0+60=245。
import java.util.*;public class Main {static final int MAX_SPEED = Integer.MAX_VALUE;enum Direction { UP, RIGHT, DOWN, LEFT };static class Point {int x, y, direction, speed;public Point(int x, int y, int direction, int speed) {this.x = x;this.y = y;this.direction = direction;this.speed = speed;}}static class Compare implements Comparator<Point> {public int compare(Point a, Point b) {return Integer.compare(a.speed, b.speed);}}static int calcTime(int[][] lights, int timePerRoad, int rowStart,int colStart, int rowEnd, int colEnd) {int[][][] result = new int[rowEnd + 1][colEnd + 1][4];for (int[][] row : result) {for (int[] col : row) {Arrays.fill(col, MAX_SPEED);}}PriorityQueue<Point> pq = new PriorityQueue<>(new Compare());for (int i = 0; i < 4; i++) {pq.add(new Point(rowStart, colStart, i, 0));}int[][] directions = {{-1, 0}, {0, 1}, {1, 0}, {0, -1}};while (!pq.isEmpty()) {Point p = pq.poll();if (p.speed > result[p.x][p.y][p.direction]) {continue;}for (int i = 0; i < 4; ++i) {int newDir = (p.direction + i) % 4;int newX = p.x + directions[newDir][0];int newY = p.y + directions[newDir][1];if (newX >= 0 && newX <= rowEnd && newY >= 0 && newY <= colEnd) {int newSpeed = p.speed + timePerRoad + (i != 1 ? lights[p.x][p.y] : 0);if (newSpeed < result[newX][newY][newDir]) {result[newX][newY][newDir] = newSpeed;pq.add(new Point(newX, newY, newDir, newSpeed));}}}}return Math.min(Math.min(result[rowEnd][colEnd][0], result[rowEnd][colEnd][1]),result[rowEnd][colEnd][2]);}public static void main(String[] args) {int[][] lights = {{1, 2, 3}, {4, 5, 6}, {7, 8, 9}};int timePerRoad = 60;int rowStart = 0;int colStart = 0;int rowEnd = 2;int colEnd = 2;System.out.println(calcTime(lights, timePerRoad, rowStart, colStart, rowEnd, colEnd));}
}
