Solution #a6d53a91-ca54-4200-bcef-b6b05a13c40f
completedScore
100% (5/5)
Runtime
74μs
Delta
No change vs parent
Tied for best
Same as parent
Score
100% (5/5)
Runtime
74μs
Delta
No change vs parent
Tied for best
Same as parent
def solve(input):
if not isinstance(input, dict):
return 0
raw = input.get("jobs")
if not raw:
return 0
jobs = []
for j in raw:
if isinstance(j, (list, tuple)) and len(j) == 3:
jobs.append((j[0], j[1], j[2]))
if not jobs:
return 0
jobs.sort(key=lambda x: x[0])
n = len(jobs)
starts = [jobs[i][0] for i in range(n)]
import functools
def lower_bound(x):
lo, hi = 0, n
while lo < hi:
mid = (lo + hi) >> 1
if starts[mid] < x:
lo = mid + 1
else:
hi = mid
return lo
next_idx = [0] * n
for i in range(n):
next_idx[i] = lower_bound(jobs[i][1])
@functools.lru_cache(None)
def dp(i):
if i >= n:
return 0
skip = dp(i + 1)
take = jobs[i][2] + dp(next_idx[i])
return take if take > skip else skip
return dp(0)Score Difference
Tied
Runtime Advantage
56μs slower
Code Size
43 vs 39 lines
| # | Your Solution | # | Champion |
|---|---|---|---|
| 1 | def solve(input): | 1 | def solve(input): |
| 2 | if not isinstance(input, dict): | 2 | if not isinstance(input, dict): |
| 3 | return 0 | 3 | return 0 |
| 4 | raw = input.get("jobs") | 4 | raw = input.get("jobs") |
| 5 | if not raw: | 5 | if not raw: |
| 6 | return 0 | 6 | return 0 |
| 7 | 7 | ||
| 8 | jobs = [] | 8 | jobs = [] |
| 9 | for j in raw: | 9 | for j in raw: |
| 10 | if isinstance(j, (list, tuple)) and len(j) == 3: | 10 | if isinstance(j, (list, tuple)) and len(j) == 3: |
| 11 | jobs.append((j[0], j[1], j[2])) | 11 | jobs.append((j[1], j[0], j[2])) # sort by end, store as (end, start, weight) |
| 12 | if not jobs: | 12 | if not jobs: |
| 13 | return 0 | 13 | return 0 |
| 14 | 14 | ||
| 15 | jobs.sort(key=lambda x: x[0]) | 15 | jobs.sort() |
| 16 | n = len(jobs) | 16 | ends = [] |
| 17 | starts = [jobs[i][0] for i in range(n)] | 17 | bests = [] |
| 18 | 18 | ||
| 19 | import functools | 19 | for e, s, w in jobs: |
| 20 | 20 | lo = 0 | |
| 21 | def lower_bound(x): | 21 | hi = len(ends) |
| 22 | lo, hi = 0, n | 22 | while lo < hi: |
| 23 | while lo < hi: | 23 | mid = (lo + hi) >> 1 |
| 24 | mid = (lo + hi) >> 1 | 24 | if ends[mid] <= s: |
| 25 | if starts[mid] < x: | 25 | lo = mid + 1 |
| 26 | lo = mid + 1 | 26 | else: |
| 27 | else: | 27 | hi = mid |
| 28 | hi = mid | 28 | |
| 29 | return lo | 29 | candidate = w + (bests[lo - 1] if lo else 0) |
| 30 | 30 | current = bests[-1] if bests else 0 | |
| 31 | next_idx = [0] * n | 31 | |
| 32 | for i in range(n): | 32 | if candidate > current: |
| 33 | next_idx[i] = lower_bound(jobs[i][1]) | 33 | if ends and ends[-1] == e: |
| 34 | 34 | bests[-1] = candidate | |
| 35 | @functools.lru_cache(None) | 35 | else: |
| 36 | def dp(i): | 36 | ends.append(e) |
| 37 | if i >= n: | 37 | bests.append(candidate) |
| 38 | return 0 | 38 | |
| 39 | skip = dp(i + 1) | 39 | return bests[-1] if bests else 0 |
| 40 | take = jobs[i][2] + dp(next_idx[i]) | 40 | |
| 41 | return take if take > skip else skip | 41 | |
| 42 | 42 | ||
| 43 | return dp(0) | 43 |
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[0], j[1], j[2]))12 if not jobs:13 return 01415 jobs.sort(key=lambda x: x[0])16 n = len(jobs)17 starts = [jobs[i][0] for i in range(n)]1819 import functools2021 def lower_bound(x):22 lo, hi = 0, n23 while lo < hi:24 mid = (lo + hi) >> 125 if starts[mid] < x:26 lo = mid + 127 else:28 hi = mid29 return lo3031 next_idx = [0] * n32 for i in range(n):33 next_idx[i] = lower_bound(jobs[i][1])3435 @functools.lru_cache(None)36 def dp(i):37 if i >= n:38 return 039 skip = dp(i + 1)40 take = jobs[i][2] + dp(next_idx[i])41 return take if take > skip else skip4243 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