Re: [閒聊] 每日leetcode

作者: dont   2024-12-13 09:09:34
2593. Find Score of an Array After Marking All Elements
## 思路
建(num, idx) 的min heap
每次pop檢查idx有沒有mark過, 沒有就+score並mark
## Code
```python
class Solution:
def findScore(self, nums: List[int]) -> int:
n = len(nums)
heap = []
for idx, num in enumerate(nums):
heapq.heappush(heap, (num, idx))
res = 0
while heap:
num, idx = heapq.heappop(heap)
if nums[idx] == num:
res += num
nums[idx] = 0
if idx: nums[idx-1] = 0
if idx < n-1: nums[idx+1] = 0
return res
```
作者: DJYOMIYAHINA (通通打死)   2023-12-13 09:09:00
別卷了
作者: Furina (芙寧娜)   2024-12-13 09:11:00
大師

Links booklink

Contact Us: admin [ a t ] ucptt.com