結果
問題 |
No.704 ゴミ拾い Medium
|
ユーザー |
|
提出日時 | 2025-07-22 17:33:07 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 749 ms / 1,500 ms |
コード長 | 3,871 bytes |
コンパイル時間 | 542 ms |
コンパイル使用メモリ | 82,404 KB |
実行使用メモリ | 208,696 KB |
最終ジャッジ日時 | 2025-07-22 17:33:28 |
合計ジャッジ時間 | 20,801 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 44 |
ソースコード
import os,sys,random,threading #sys.exit() 退出程序 #sys.setrecursionlimit(10**6) #调整栈空间 from random import randint,choice,shuffle #randint(a,b)从[a,b]范围随机选择一个数 #choice(seq)seq可以是一个列表,元组或字符串,从seq中随机选取一个元素 #shuffle(x)将一个可变的序列x中的元素打乱 from copy import deepcopy from io import BytesIO,IOBase from types import GeneratorType from functools import lru_cache,reduce #reduce(op,迭代对象) from bisect import bisect_left,bisect_right #bisect_left(x) 大于等于x的第一个下标 #bisect_right(x) 大于x的第一个下标 from collections import Counter,defaultdict,deque from itertools import accumulate,combinations,permutations #accumulate(a)用a序列生成一个累积迭代器,一般list化前面放个[0]做前缀和用 #combinations(a,k)a序列选k个 组合迭代器 #permutations(a,k)a序列选k个 排列迭代器 from heapq import heapify,heappop,heappush #heapify将列表转为堆 from typing import Generic,Iterable,Iterator,TypeVar,Union,List from string import ascii_lowercase,ascii_uppercase,digits #小写字母,大写字母,十进制数字 from math import ceil,floor,sqrt,pi,factorial,gcd,log,log10,log2,inf #ceil向上取整,floor向下取整 ,sqrt开方 ,factorial阶乘 from decimal import Decimal,getcontext #Decimal(s) 实例化Decimal对象,一般使用字符串 #getcontext().prec=100 修改精度 from sys import stdin, stdout, setrecursionlimit input = lambda: sys.stdin.readline().rstrip("\r\n") MI = lambda :map(int,input().split()) li = lambda :list(MI()) ii = lambda :int(input()) mod = int(1e9 + 7) #998244353 inf = 1<<60 py = lambda :print("YES") pn = lambda :print("NO") DIRS = [(0, 1), (1, 0), (0, -1), (-1, 0)] # 右下左上 DIRS8 = [(0, 1), (1, 1), (1, 0), (1, -1), (0, -1), (-1, -1), (-1, 0),(-1, 1)] # →↘↓↙←↖↑↗ class ZKW: """自低向上非递归写法线段树,0_indexed tmx = ZKW(pre, max, -2 ** 61) """ __slots__ = ('n', 'op', 'e', 'log', 'size', 'd') def __init__(self, n, OP, E): """ OP: 操作:max,min,sum E: 每个元素默认值 """ self.n = n self.op = OP self.e = E self.log = (self.n - 1).bit_length() self.size = 1 << self.log self.d = [E for i in range(2 * self.size)] def build(self, V): for i in range(self.n): self.d[self.size+i]=V[i] for i in range(self.size-1,0,-1): self.update(i) def set(self, p, x): # assert 0 <= p and p < self.n update = self.update p += self.size self.d[p] = x for i in range(1, self.log + 1): update(p >> i) def get(self,p): p+=self.size return self.d[p] def query(self, l, r): # [l,r] 双闭区间 # assert 0 <= l and l <= r and r <= self.n sml, smr, op, d = self.e, self.e, self.op, self.d l += self.size r += self.size+1 while l < r: if l & 1: sml = op(sml, d[l]) l += 1 if r & 1: smr = op(d[r - 1], smr) r -= 1 l >>= 1 r >>= 1 return self.op(sml, smr) def update(self, k): self.d[k] = self.op(self.d[2 * k], self.d[2 * k + 1]) def all_query(self): return self.d[1] n=ii() a=[0]+li() x=[0]+li() y=[0]+li() d=sorted(set(a+x)) d={x:i for i,x in enumerate(d)} s1=ZKW(len(d),min,inf) s2=ZKW(len(d),min,inf) dp=[inf]*(n+1) dp[0]=0 for i in range(1,n+1): idx=d[x[i]] if dp[i-1]+x[i]+y[i]<s1.get(idx): s1.set(idx,dp[i-1]+x[i]+y[i]) if dp[i-1]-x[i]+y[i]<s2.get(idx): s2.set(idx,dp[i-1]-x[i]+y[i]) idx=d[a[i]] dp[i]=min(s2.query(0,idx)+a[i],s1.query(idx,len(d)-1)-a[i]) print(dp[n])