Medium
You are given two integers, n
and k
, along with two 2D integer arrays, stayScore
and travelScore
.
A tourist is visiting a country with n
cities, where each city is directly connected to every other city. The tourist’s journey consists of exactly k
0-indexed days, and they can choose any city as their starting point.
Each day, the tourist has two choices:
curr
during day i
, they will earn stayScore[i][curr]
points.curr
to city dest
, they will earn travelScore[curr][dest]
points.Return the maximum possible points the tourist can earn.
Example 1:
Input: n = 2, k = 1, stayScore = [[2,3]], travelScore = [[0,2],[1,0]]
Output: 3
Explanation:
The tourist earns the maximum number of points by starting in city 1 and staying in that city.
Example 2:
Input: n = 3, k = 2, stayScore = [[3,4,2],[2,1,2]], travelScore = [[0,2,1],[2,0,4],[3,2,0]]
Output: 8
Explanation:
The tourist earns the maximum number of points by starting in city 1, staying in that city on day 0, and traveling to city 2 on day 1.
Constraints:
1 <= n <= 200
1 <= k <= 200
n == travelScore.length == travelScore[i].length == stayScore[i].length
k == stayScore.length
1 <= stayScore[i][j] <= 100
0 <= travelScore[i][j] <= 100
travelScore[i][i] == 0
public class Solution {
public int maxScore(int n, int k, int[][] stayScores, int[][] travelScores) {
// dp[day][city]
int[][] dp = new int[k + 1][n];
int result = 0;
for (int day = k - 1; day >= 0; day--) {
for (int city = 0; city < n; city++) {
int stayScore = stayScores[day][city] + dp[day + 1][city];
int travelScore = 0;
for (int nextCity = 0; nextCity < n; nextCity++) {
int nextScore = travelScores[city][nextCity] + dp[day + 1][nextCity];
travelScore = Math.max(nextScore, travelScore);
}
dp[day][city] = Math.max(stayScore, travelScore);
result = Math.max(dp[day][city], result);
}
}
return result;
}
}