解法洛谷P3406 海底高铁 python解法

最新推荐文章于 2024-01-19 18:18:14 发布

原创最新推荐文章于 2024-01-19 18:18:14 发布 · 254 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#算法 #数据结构 #python

该文章介绍了一种使用Python实现一维差分的方法，通过静态方法`insert`更新差分数组`t`，并处理边界条件。在解决方案中，对输入数据进行处理，根据条件插入差分值，最后计算并输出特定条件下的总和。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

这里采用一维差分的方法，用python实现，代码如下：
"""差分是前缀和的逆过程，a[i]=sum(b1,b2,b3.....bi)。通过bi+1，a[i]之后的都会加一"""

t = [0]*100010
class Solution():
    @staticmethod
    def insert(l,r,c):  #由于初始值都是0就省去了两个列表的转化
        t[l] += c
        t[r] -= c

    def solution(self):
        n,m = map(int,input().split())
        k=list(map(int,input().split()))
        for i in range(1,m):
            if k[i-1] > k[i]:
                Solution.insert(k[i],k[i-1],1)
            else:
                Solution.insert(k[i-1],k[i],1)
        for i in range(1,n):
            t[i] += t[i-1]
        sum = 0
        for i in range(1,n):
            a,b,c = map(int,input().split())
            if a*t[i] > c+b*t[i]:
                sum += (c+b*t[i])
            else:
                sum += a*t[i]
        print(sum)
s=Solution()
s.solution()