Solution #7cbd604b-f218-4e16-8eed-b65017e809b8
completedScore
80% (4/5)
Runtime
46μs
Delta
New score
-20.0% vs best
Improved from parent
Score
80% (4/5)
Runtime
46μs
Delta
New score
-20.0% vs best
Improved from parent
def solve(input):
jobs = input.get("jobs", [])
if not jobs:
return 0
# Recursive weighted interval scheduling:
# sort by start time, then for each job choose skip/take.
# If taking a job, jump to the first job with start >= current end.
jobs = sorted((s, e, w) for s, e, w in jobs)
n = len(jobs)
starts = [s for s, _, _ in jobs]
next_idx = [n] * n
for i in range(n):
end_i = jobs[i][1]
lo, hi = i + 1, n
while lo < hi:
mid = (lo + hi) // 2
if starts[mid] < end_i:
lo = mid + 1
else:
hi = mid
next_idx[i] = lo
memo = {}
def dp(i):
if i >= n:
return 0
if i in memo:
return memo[i]
skip = dp(i + 1)
take = jobs[i][2] + dp(next_idx[i])
ans = take if take > skip else skip
memo[i] = ans
return ans
return dp(0)Score Difference
-20.0%
Runtime Advantage
28μs slower
Code Size
38 vs 39 lines
| # | Your Solution | # | Champion |
|---|---|---|---|
| 1 | def solve(input): | 1 | def solve(input): |
| 2 | jobs = input.get("jobs", []) | 2 | if not isinstance(input, dict): |
| 3 | if not jobs: | 3 | return 0 |
| 4 | return 0 | 4 | raw = input.get("jobs") |
| 5 | 5 | if not raw: | |
| 6 | # Recursive weighted interval scheduling: | 6 | return 0 |
| 7 | # sort by start time, then for each job choose skip/take. | 7 | |
| 8 | # If taking a job, jump to the first job with start >= current end. | 8 | jobs = [] |
| 9 | jobs = sorted((s, e, w) for s, e, w in jobs) | 9 | for j in raw: |
| 10 | n = len(jobs) | 10 | if isinstance(j, (list, tuple)) and len(j) == 3: |
| 11 | starts = [s for s, _, _ in jobs] | 11 | jobs.append((j[1], j[0], j[2])) # sort by end, store as (end, start, weight) |
| 12 | 12 | if not jobs: | |
| 13 | next_idx = [n] * n | 13 | return 0 |
| 14 | for i in range(n): | 14 | |
| 15 | end_i = jobs[i][1] | 15 | jobs.sort() |
| 16 | lo, hi = i + 1, n | 16 | ends = [] |
| 17 | while lo < hi: | 17 | bests = [] |
| 18 | mid = (lo + hi) // 2 | 18 | |
| 19 | if starts[mid] < end_i: | 19 | for e, s, w in jobs: |
| 20 | lo = mid + 1 | 20 | lo = 0 |
| 21 | else: | 21 | hi = len(ends) |
| 22 | hi = mid | 22 | while lo < hi: |
| 23 | next_idx[i] = lo | 23 | mid = (lo + hi) >> 1 |
| 24 | 24 | if ends[mid] <= s: | |
| 25 | memo = {} | 25 | lo = mid + 1 |
| 26 | 26 | else: | |
| 27 | def dp(i): | 27 | hi = mid |
| 28 | if i >= n: | 28 | |
| 29 | return 0 | 29 | candidate = w + (bests[lo - 1] if lo else 0) |
| 30 | if i in memo: | 30 | current = bests[-1] if bests else 0 |
| 31 | return memo[i] | 31 | |
| 32 | skip = dp(i + 1) | 32 | if candidate > current: |
| 33 | take = jobs[i][2] + dp(next_idx[i]) | 33 | if ends and ends[-1] == e: |
| 34 | ans = take if take > skip else skip | 34 | bests[-1] = candidate |
| 35 | memo[i] = ans | 35 | else: |
| 36 | return ans | 36 | ends.append(e) |
| 37 | 37 | bests.append(candidate) | |
| 38 | return dp(0) | 38 | |
| 39 | 39 | return bests[-1] if bests else 0 |
1def solve(input):2 jobs = input.get("jobs", [])3 if not jobs:4 return 056 # Recursive weighted interval scheduling:7 # sort by start time, then for each job choose skip/take.8 # If taking a job, jump to the first job with start >= current end.9 jobs = sorted((s, e, w) for s, e, w in jobs)10 n = len(jobs)11 starts = [s for s, _, _ in jobs]1213 next_idx = [n] * n14 for i in range(n):15 end_i = jobs[i][1]16 lo, hi = i + 1, n17 while lo < hi:18 mid = (lo + hi) // 219 if starts[mid] < end_i:20 lo = mid + 121 else:22 hi = mid23 next_idx[i] = lo2425 memo = {}2627 def dp(i):28 if i >= n:29 return 030 if i in memo:31 return memo[i]32 skip = dp(i + 1)33 take = jobs[i][2] + dp(next_idx[i])34 ans = take if take > skip else skip35 memo[i] = ans36 return ans3738 return dp(0)1def solve(input):2 if not isinstance(input, dict):3 return 04 raw = input.get("jobs")5 if not raw:6 return 078 jobs = []9 for j in raw:10 if isinstance(j, (list, tuple)) and len(j) == 3:11 jobs.append((j[1], j[0], j[2])) # sort by end, store as (end, start, weight)12 if not jobs:13 return 01415 jobs.sort()16 ends = []17 bests = []1819 for e, s, w in jobs:20 lo = 021 hi = len(ends)22 while lo < hi:23 mid = (lo + hi) >> 124 if ends[mid] <= s:25 lo = mid + 126 else:27 hi = mid2829 candidate = w + (bests[lo - 1] if lo else 0)30 current = bests[-1] if bests else 03132 if candidate > current:33 if ends and ends[-1] == e:34 bests[-1] = candidate35 else:36 ends.append(e)37 bests.append(candidate)3839 return bests[-1] if bests else 0