【代码随想录算法训练营第六十六天|卡码网97.小明逛公园、127.骑士的攻击】

文章目录

97.小明逛公园

之前的题目都是只有一个出发点和到达点,这道题是有多个起始对,用之前的算法把每一对结果算出来也是可行的,在这里使用Floyd算法。

本质上是一种动态规划,dp数组dp[i][j][k]中的i,j两点在以(1~k)中的点作为中间结点的时候的最小距离,递推公式是dp[i][j][k]=min(dp[i][k][k-1]+dp[k][j][k-1], dp[i][j][k-1]),遍历顺序是k在最外,i,j在内。本质上是因为dp[i][k][k-1]这些中间量还可能是由k更小的时候的组合推断出来的,所以对于所有的i和j,k都应该是从小到大递增推导所有i和j的状态,所以k在最外层。

python 复制代码
n, m = map(int, input().split())
grid = [[[float('inf')] * (n+1) for _ in range(n+1)] for _ in range(n+1)]
for i in range(m):
    u, v, w = map(int, input().split())
    grid[u][v][0] = w
    grid[v][u][0] = w 
q = int(input())
plans = []
for i in range(q):
    start, end = map(int, input().split())
    plans.append([start, end])
    
for k in range(1, n+1):
    for i in range(1, n+1):
        for j in range(1, n+1):
            grid[i][j][k] = min(grid[i][j][k-1], grid[i][k][k-1] + grid[k][j][k-1])


for plan in plans:
    if grid[plan[0]][plan[1]][-1] != float('inf'):
        print(grid[plan[0]][plan[1]][-1])
    else:
        print(-1)

127.骑士的攻击

这道题使用的是Astar算法,是在BFS的基础上进行了一定的优化。普通的BFS也能做但是在广度搜索的时候会有很多的浪费,因此关键就在于每次如何选择朝向目标移动的最近的点。Astar通过计算每个点的预估距离并用小顶堆进行排序,从而使得每次取出来的都是当前可能最小距离的点。在这题使用欧氏距离最合适,总距离是点距离原点的欧氏距离加上到目标点的欧氏距离。

python 复制代码
import heapq

n = int(input())
directions = [[-2, 1], [-1, 2], [1, 2], [2, 1], [-2, -1], [-1, -2], [1, -2], [2, -1]]
Distance = lambda x, y, tar_x, tar_y: (x - tar_x) ** 2 + (y - tar_y) ** 2

class Knight:
    def __init__(self, a1, a2, g=0, f=0, h=0):
        self.x = a1
        self.y = a2
        self.g = g
        self.f = f 
        self.h = h
    
    def __lt__(self, k):
        return self.f < k.f
    
    def __str__(self):
        return f'Knight:{self.x},{self.y}'
    
def bfs(grid, knight, target):
    heap = []
    heapq.heappush(heap, knight)
    while heap:
        cur = heapq.heappop(heap)
        if cur.x == target[0] and cur.y == target[1]:
            return
        for d in directions:
            next_x = cur.x + d[0]
            next_y = cur.y + d[1]
            if next_x < 1 or next_x >= len(grid) or next_y < 1 or next_y >= len(grid[0]):
                continue
            if not grid[next_x][next_y]:
                grid[next_x][next_y] = grid[cur.x][cur.y] + 1
                next_knight = Knight(next_x, next_y)
                next_knight.g = cur.g + 5
                next_knight.h = Distance(next_x, next_y, target[0], target[1])
                next_knight.f = next_knight.g + next_knight.h 
                heapq.heappush(heap, next_knight)
                
for i in range(n):
    a1, a2, b1, b2 = map(int, input().split())
    grid = [[0] * 1001 for _ in range(1001)]
    knight = Knight(a1, a2)
    knight.h = Distance(a1, a2, b1, b2)
    knight.f = knight.g + knight.h
    bfs(grid, knight, [b1, b2])
    print(grid[b1][b2])
相关推荐
想跑步的小弱鸡1 小时前
Leetcode hot 100(day 3)
算法·leetcode·职场和发展
xyliiiiiL3 小时前
ZGC初步了解
java·jvm·算法
爱的叹息3 小时前
RedisTemplate 的 6 个可配置序列化器属性对比
算法·哈希算法
独好紫罗兰4 小时前
洛谷题单2-P5713 【深基3.例5】洛谷团队系统-python-流程图重构
开发语言·python·算法
每次的天空5 小时前
Android学习总结之算法篇四(字符串)
android·学习·算法
请来次降维打击!!!5 小时前
优选算法系列(5.位运算)
java·前端·c++·算法
qystca5 小时前
蓝桥云客 刷题统计
算法·模拟
别NULL5 小时前
机试题——统计最少媒体包发送源个数
c++·算法·媒体
weisian1516 小时前
Java常用工具算法-3--加密算法2--非对称加密算法(RSA常用,ECC,DSA)
java·开发语言·算法
程序员黄同学7 小时前
贪心算法,其优缺点是什么?
算法·贪心算法