dfs and similar dp graphs shortest paths *2400

Please click on ads to support us..

Python Code:

from collections import deque
import sys, os, io
input = io.BytesIO(os.read(0, os.fstat(0).st_size)).readline

def f(u, v):
    return u * g + v

n, m = map(int, input().split())
d = list(map(int, input().split()))
d.sort()
g, r = map(int, input().split())
inf = pow(10, 9) + 1
l0, r0 = [inf] * m, [inf] * m
for i in range(m - 1):
    d0 = d[i + 1] - d[i]
    l0[i + 1], r0[i] = d0, d0
dp = [inf] * (m * g)
dp[0] = 0
q = deque([0])
c = 0
while q:
    v = q.popleft()
    i, j = divmod(v, g)
    for ni, nj in [(i - 1, j + l0[i]), (i + 1, j + r0[i])]:
        u = f(ni, nj % g)
        if nj < g and dp[u] == inf:
            dp[u] = dp[v]
            q.appendleft(u)
        elif nj == g and dp[u] == inf:
            dp[u] = dp[v] + 1
            q.append(u)
ans = inf
for i in range(m - 1, -1, -1):
    if d[i] + g < n:
        break
    ans = min(ans, dp[f(i, 0)] * (g + r) + n - d[i])
ans = (ans + 1) % (inf + 1) - 1
print(ans)


Comments

Submit
0 Comments
More Questions

1516A - Tit for Tat
622. Design Circular Queue
814. Binary Tree Pruning
791. Custom Sort String
787. Cheapest Flights Within K Stops
779. K-th Symbol in Grammar
701. Insert into a Binary Search Tree
429. N-ary Tree Level Order Traversal
739. Daily Temperatures
647. Palindromic Substrings
583. Delete Operation for Two Strings
518. Coin Change 2
516. Longest Palindromic Subsequence
468. Validate IP Address
450. Delete Node in a BST
445. Add Two Numbers II
442. Find All Duplicates in an Array
437. Path Sum III
436. Find Right Interval
435. Non-overlapping Intervals
406. Queue Reconstruction by Height
380. Insert Delete GetRandom O(1)
332. Reconstruct Itinerary
368. Largest Divisible Subset
377. Combination Sum IV
322. Coin Change
307. Range Sum Query - Mutable
287. Find the Duplicate Number
279. Perfect Squares
275. H-Index II