力扣 第 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

class Solution:
    def maxPathScore(self, g: List[List[int]], k: int) -> int:
        n,m=len(g),len(g[0])
        k=min(k,m+n-1)
        f=[[[-inf]*(k+1) for i in range(m)] for j in range(n)]
        # print(f)
        cost=0 if g[0][0]==0 else 1
        if cost>k:
            return -1
        f[0][0][cost]=g[0][0]
        for i in range(n):
            for j in range(m):
                if i==0 and j==0:
                    continue
                for u in range(k+1):
                    maxx=-inf
                    if i>0:
                        preu=u-(1 if g[i][j] else 0)
                        if preu>=0:
                            maxx=max(maxx,f[i-1][j][preu]+g[i][j])
                    if j>0:
                        preu=u-(1 if g[i][j] else 0)
                        if preu>=0:
                            maxx=max(maxx,f[i][j-1][preu]+g[i][j])
                    if maxx!=-inf:
                        f[i][j][u]=maxx
        ans=-1
        for i in range(k+1):
            ans=max(ans,f[n-1][m-1][i])
        return ans if ans!=-inf else -1