力扣 第 475 场周赛(A~C)

A:3740. 三个相等元素之间的最小距离 I
见B。

B:3741. 三个相等元素之间的最小距离 II

给定数组nums，要求值相同的三元组的下标最小距离，距离定义为abs(i - j) + abs(j - k) + abs(k - i)，数据范围为1e5.

我们我们可以按[值，下标]升序排序，然后遍历值相同的三个元素，来更新答案。

class Solution:
    def minimumDistance(self, nums: List[int]) -> int:
        p=[]
        for i,x in enumerate(nums):
            p.append([x,i])
        p.sort(key=lambda x:[x[0],x[1]])
        ans=inf
        for i in range(2,len(p)):
            if p[i][0]==p[i-1][0] and p[i][0]==p[i-2][0]:
                a=p[i][1]
                b=p[i-1][1]
                c=p[i-2][1]
                ans=min(ans,abs(a-b)+abs(a-c)+abs(c-b))
        return ans if ans!=inf else -1

C:3742. 网格中得分最大的路径

上来题读错了，看成四个方向了，其实是右和下的递推。

f[i][j][k]表示走到i,j但是花费不超过k的最大分数，则有f[ i ][ j ][ k ]=max(f[ i-1 ][ j ][ u ], f[ i ][ j-1 ][ u ])，其中u=k-1 if g[i][j]>0 else k.

起始条件为f[i][j][cost]=0,其余为-inf，其中cost=0 if g[0][0]==0 else 1

// const int N = 210;
// int f[N][N][2*N];
class Solution {
public:
    int maxPathScore(vector<vector<int>>& g, int d) {
        int n=g.size();
        int m=g[0].size();
        d=min(d,m+n-1);
        vector<vector<vector<int>>> f(n, vector<vector<int>>(m, vector<int>(d + 1, -0x3f3f3f3f)));
        // memset(f,-0x3f,sizeof f);
        int cost = (g[0][0]==0?0:1);
        // if(cost>d)
        //     return -1;
        f[0][0][cost]=0;
        for(int i=0;i<n;i++){
            for(int j=0;j<m;j++){
                // if(i==0 && j==0) continue;
                for(int k=0;k<=d;k++){
                    if(i>0){
                        int prek=k;
                        if(g[i][j]!=0)
                            prek-=1;
                        if(prek>=0)
                            f[i][j][k]=max(f[i][j][k],f[i-1][j][prek]+g[i][j]);
                    }
                    if(j>0){
                        int prek=k;
                        if(g[i][j]!=0)
                            prek-=1;
                        if(prek>=0)
                            f[i][j][k]=max(f[i][j][k],f[i][j-1][prek]+g[i][j]);
                    }
                }
            }
        }
        int ans=-1;
        for(int i=0;i<=d;i++){
            ans=max(ans,f[n-1][m-1][i]);
        }
        return ans;
    }
};