Re: [閒聊] 每日leetcode已回收

看板Marginalman作者DJYOMIYAHINA (通通打死)時間1年前 (2024/07/13 11:56)推噓4(4推 0噓 4→)

留言8則, 5人參與討論串486/1553 (看更多)

蝦J8亂寫最後zip unpack那邊因為空list會出錯搞了好久== 就先這樣了應該很多地方可以優化 def survivedRobotsHealths(self, positions: List[int], healths: List[int], directions: str) -> List[int]: n = len(positions) idx = [i+1 for i in range(n)] positions, healths, directions, idx = zip(*sorted(zip(positions, healths, directions, idx))) positions, healths, directions, idx = list(positions), list(healths), list(directions), list(idx) print(positions) stk = [] ans = [] for i in range(n): if directions[i] == "R": stk.append(i) else: while stk: if healths[stk[-1]] < healths[i]: stk.pop() healths[i] -= 1 elif healths[stk[-1]] == healths[i]: stk.pop() healths[i] = 0 break else: healths[stk[-1]] -= 1 break if not stk and healths[i] > 0: ans.append(i) ans = ans + stk ans = [(idx[i], healths[i]) for i in ans] if ans: _, ans = zip(*sorted(ans)) return list(ans) else: return [] -- ※ 發信站: 批踢踢實業坊(ptt.cc), 來自: 125.228.146.144 (臺灣) ※ 文章網址: https://www.ptt.cc/bbs/Marginalman/M.1720843000.A.293.html

推

nozomizo

07/13 11:58, 1年前 , 1^F

07/13 11:58, 1^F

推

a114514

07/13 12:00, 1年前 , 2^F

07/13 12:00, 2^F

推

oin1104

07/13 12:00, 1年前 , 3^F

07/13 12:00, 3^F