- Define precisely what one DP state represents.
- Establish the base cases before transitions are evaluated.
- Process states in dependency order and combine only already-known values.
Code notes
- 65 lines of Java from the credited upstream file 3414.java.
- The implementation visibly relies on sequence storage, cached states.
- 3 loop blocks detected.
Complexity
Multiply the number of reachable states by the work performed for each transition, then include the stored state table in memory usage.
Check the problem constraints before deciding whether this complexity will pass.
Use this to learn the idea, then write your own version.
1class Solution {2 public int[] maximumWeight(List<List<Integer>> intervals) {3 4 List<Interval> indexedIntervals = new ArrayList<>();5 for (int i = 0; i < intervals.size(); ++i) {6 List<Integer> interval = intervals.get(i);7 indexedIntervals.add(new Interval(interval.get(0), interval.get(1), interval.get(2), i));8 }9 indexedIntervals.sort(Comparator.comparingInt(Interval::left));10 T[][] memo = new T[indexedIntervals.size()][5];11 return dp(indexedIntervals, memo, 0, 4).selected.stream().mapToInt(Integer::intValue).toArray();12 }13 14 private record T(long weight, List<Integer> selected) {}15 private record Interval(int left, int right, int weight, int originalIndex) {}16 17 private T dp(List<Interval> intervals, T[][] memo, int i, int quota) {18 if (i == intervals.size() || quota == 0)19 return new T(0, List.of());20 if (memo[i][quota] != null)21 return memo[i][quota];22 23 T skip = dp(intervals, memo, i + 1, quota);24 25 Interval interval = intervals.get(i);26 final int j = findFirstGreater(intervals, i + 1, interval.right);27 T nextRes = dp(intervals, memo, j, quota - 1);28 29 List<Integer> newSelected = new ArrayList<>(nextRes.selected);30 newSelected.add(interval.originalIndex);31 Collections.sort(newSelected);32 T pick = new T(interval.weight + nextRes.weight, newSelected);33 return memo[i][quota] =34 (pick.weight > skip.weight ||35 (pick.weight == skip.weight && compareLists(pick.selected, skip.selected) < 0))36 ? pick37 : skip;38 }39 40 41 private int findFirstGreater(List<Interval> intervals, int startFrom, int rightBoundary) {42 int l = startFrom;43 int r = intervals.size();44 while (l < r) {45 final int m = (l + r) / 2;46 if (intervals.get(m).left > rightBoundary)47 r = m;48 else49 l = m + 1;50 }51 return l;52 }53 54 55 private int compareLists(List<Integer> list1, List<Integer> list2) {56 final int minSize = Math.min(list1.size(), list2.size());57 for (int i = 0; i < minSize; ++i) {58 final int comparison = Integer.compare(list1.get(i), list2.get(i));59 if (comparison != 0)60 return comparison;61 }62 return Integer.compare(list1.size(), list2.size());63 }64}65