【leetcode】二指输入的的最小距离 (dp)

二指输入法定制键盘在 XY 平面上的布局如上图所示，其中每个大写英文字母都位于某个坐标处，例如字母 A 位于坐标 (0,0)，字母 B 位于坐标 (0,1)，字母 P 位于坐标 (2,3) 且字母 Z 位于坐标 (4,1)。

给你一个待输入字符串 word，请你计算并返回在仅使用两根手指的情况下，键入该字符串需要的最小移动总距离。坐标 (x1,y1) 和 (x2,y2) 之间的距离是 |x1 - x2| + |y1 - y2|。

注意，两根手指的起始位置是零代价的，不计入移动总距离。你的两根手指的起始位置也不必从首字母或者前两个字母开始。

示例 1：

输入：word = “CAKE”
输出：3
解释：
使用两根手指输入 “CAKE” 的最佳方案之一是：
手指 1 在字母 ‘C’ 上 -> 移动距离 = 0
手指 1 在字母 ‘A’ 上 -> 移动距离 = 从字母 ‘C’ 到字母 ‘A’ 的距离 = 2
手指 2 在字母 ‘K’ 上 -> 移动距离 = 0
手指 2 在字母 ‘E’ 上 -> 移动距离 = 从字母 ‘K’ 到字母 ‘E’ 的距离 = 1
总距离 = 3
示例 2：

输入：word = “HAPPY”
输出：6
解释：
使用两根手指输入 “HAPPY” 的最佳方案之一是：
手指 1 在字母 ‘H’ 上 -> 移动距离 = 0
手指 1 在字母 ‘A’ 上 -> 移动距离 = 从字母 ‘H’ 到字母 ‘A’ 的距离 = 2
手指 2 在字母 ‘P’ 上 -> 移动距离 = 0
手指 2 在字母 ‘P’ 上 -> 移动距离 = 从字母 ‘P’ 到字母 ‘P’ 的距离 = 0
手指 1 在字母 ‘Y’ 上 -> 移动距离 = 从字母 ‘A’ 到字母 ‘Y’ 的距离 = 4
总距离 = 6
示例 3：

输入：word = “NEW”
输出：3
示例 4：

输入：word = “YEAR”
输出：7

提示：

2 <= word.length <= 300
每个 word[i] 都是一个大写英文字母。

链接：https://leetcode-cn.com/problems/minimum-distance-to-type-a-word-using-two-fingers

思路分析：

对于这道题来说，它的状态和决策是比较明显的。状态有3个：字符串位置，指1坐标，指2坐标。不过**我们不可能将坐标直接作为状态，因此我们需要映射。**这里我定义了一个 int -> vector< int > 的映射idx. 并且再定义一个 char -> int 的映射mp.
于是，dp[k][i][j] 表示在前k个字符位置，指1在 i 对应坐标处，指2在 j 对应座标处所得到的最小距离。
令 id 为word[k-1]，获取当前字符所在位置，dis1, dis2表示 i 到 id, j 到 id 曼哈顿距离。
这时有两个决策：

1. 指1移到 id：dp[k][i][j] = dp[k-1][id][j] + dis1.
2. 指2移到 id：dp[k][i][j] = dp[k-1][i][id] + dis2.

上面两个决策取最小值即可。
当 k == n 时，对所有的dp取最小值，即为答案。

初始条件很明显了，k = 0 时，dp[0][i][j] = 0.

class Solution {
public:
    int minimumDistance(string word) {
        int n = word.length();
        int ans = 0x3f3f3f3f;
        map<char,int> mp;
        map<int,vector<int>> idx;
        for(int i = 0;i < 26;i++)	//初始化
        {
            char c = 'A'+i;
            mp[c] = i+1;
            int x = i/6, y = i%6;
            idx[i+1] = {x,y};
        }
        int dp[27][27][n+2];
        memset(dp,0,sizeof(dp));
        for(int k = 1;k <= n;k++)
        {
            int id = mp[word[k-1]];
            for(int i = 1;i <= 26;i++)
            {
                for(int j = 1;j <= 26;j++)
                {
                    int x1 = idx[i][0], y1 = idx[i][1];
                    int x2 = idx[j][0], y2 = idx[j][1];
                    int xx = idx[id][0], yy = idx[id][1];
                    int dis1 = abs(x1-xx) + abs(y1-yy);
                    int dis2 = abs(x2-xx) + abs(y2-yy);
                    dp[i][j][k] = min(dp[id][j][k-1]+dis1, dp[i][id][k-1]+dis2);
                    if(k == n) ans = min(ans,dp[i][j][k]);
                }
            }
        }
        return ans;
    }
};