Hello,

During my practice, I wanted to solve the problem Div 2 343, D: Babaei and Birthday Cake. In summary, we are given a sequence *A* = *a*_{1}, ..., *a*_{n} of *n* positive numbers and we should find a strictly increasing subsequence of this sequence with maximum sum. A straightforward dynamic programming solution defines a state *f*(*i*) as the maximum sum of an increasing subsequence ending at element *i*. *f*(0) = 0 and *f*(*i*) = *max*_{1 ≤ j < i}*f*(*j*) + *a*_{i} for 1 ≤ *i* ≤ *n*. The editorial explains how to implement this dynamic programming solution using segment trees in O(n log n) time.

Another solution that is also well known is to use binary search with a little modified dp recurrence. One such recurrence is to define *f*(*i*) as the maximum sum of a strictly increasing subsequence whose end is at most *a*_{i}. One can consider the elements in the order of their appearance. For element *i*, the algorithm finds the largest previous element *v* such that *a*_{v} < *a*_{i} (can be done using std::set::lower_bound) and we would have *f*(*i*) = *f*(*v*) + *a*_{i}. We then remove all elements *f*(*k*) where *a*_{k} > *a*_{i}, *k* < *i* and *f*(*k*) ≤ *f*(*i*).

I implemented the solution in C++ here. I tried to implement the solution in Java but I was not able to find a good equivalent of the std::set::lower_bound. The equivalent I know about is the ceiling method in TreeSet class however this method returns the value of the element other than an iterator to it. I wonder whether anybody knows an equivalent to std::set::lower_bound in Java.

Sorry for the long post :)

Auto comment: topic has been updated by maradonah (previous revision, new revision, compare).tailSet method

own implementation of

`lower_bound`

and`upper_bound`

works faster and takes less memoryThat's really what I need.

Thanks for highlighting it and sorry for missing it although being in the docs :)

I tried this method and it works reasonably fast. The runtime of my new java submission is even slightly better than the one based in C++.