Can this be done in O(n^2) space and O(n^3) time? Please share your approach.

# | User | Rating |
---|---|---|

1 | tourist | 3581 |

2 | OO0OOO00O0OOO0O0…O | 3264 |

3 | mnbvmar | 3246 |

4 | LHiC | 3183 |

5 | Um_nik | 3176 |

6 | Petr | 3161 |

7 | Radewoosh | 3128 |

8 | CongLingDanPaiSh…5 | 3116 |

9 | ainta | 3115 |

9 | ko_osaga | 3115 |

# | User | Contrib. |
---|---|---|

1 | Radewoosh | 189 |

2 | Errichto | 165 |

3 | rng_58 | 161 |

4 | tourist | 158 |

5 | Vovuh | 150 |

5 | Um_nik | 150 |

7 | Petr | 149 |

7 | Swistakk | 149 |

9 | 300iq | 148 |

10 | PikMike | 147 |

10 | neal | 147 |

Can this be done in O(n^2) space and O(n^3) time? Please share your approach.

↑

↓

Codeforces (c) Copyright 2010-2018 Mike Mirzayanov

The only programming contests Web 2.0 platform

Server time: Oct/21/2018 23:09:14 (f1).

Desktop version, switch to mobile version.

Supported by

User lists

Name |
---|

Auto comment: topic has been updated by ankeshgupta007 (previous revision, new revision, compare).First, you need to sort the values in non-decreasing order.

In this question, first of all, to think of a solution that will require O(n^3) space and O(n^4) time is actually pretty easy.

The States would be i,j,k where dp[i][j][k] would represent the min value of arranging the numbers from index i to j in a BST such that the root node is at a depth k. Now to calculate the value of dp[i][j][k] we will iterate from index i to j and consider one by one each index as the root. We can easily think of the recurrence for this. So I am assuming you already know that part.

Now in the above approach, we are keeping the depth also as a state. To reduce the complexity by a factor of n we can avoid keeping depth as a state. I will roughly explain my approach the main idea is if you have the value wi * (hi ^ 2) for any index i, when you will increase the depth by 1 the new value would be something like this wi * ((hi + 1)^2) which can be derived from the previous value directly. All you need is some extra values while calculating the new values,

wi * ((hi + 1)^2) = wi*(hi^2 + 1 + 2*hi)

which is wi*(hi^2) + wi + 2*wi*hi.

So the extra values which we need to store is summation of wi and summation of wi*hi. And you can get the new value which is wi * ((hi + 1)^2). And its also easy to get the new value of wi*(hi + 1). Because that would be wi*hi + wi and both the values are already there.

PS: I know I have not provided a good explanation but I think if you have the idea of the straight forward dp of this question you will get the idea of optimization from my explanation.

I understand your approach. But what is doubtful is that, when I am updating my DP tables, I am optimizing wi*(hi^2) rather than wi*(hi^2)+2*wi*hi +wi.

What I am trying to clarify is that are we making any

implicitassumption that minimizingwi*(hi^2)simultaneously minimizes wi*(hi^2)+2*wi*hi +wi (because thats the term that is included in further calculations)PS: Sorry if I am thinking something wrong.

Here is the code, I have not added memoization. It's just the recursion part. I got your doubt actually you need to look at the fact that we want to optimize only wi*(hi^2), the other values need not be optimized. We only want the other values to calculate the new values of the given function. I haven't tested my code on any test case, so do point out if you find any mistake in the implementation.

Implementation isn't a problem. Problem is when you calculate all 3 DP table for range {i,...,j} , and use the same for k=j+1 in range {i,...,j,j+1,..}, you actually include

wi*(hi^2)+2*wi*hi +wiand notwi*(hi^2). Thats where the subtlety lies.Oh okay, I got the mistake. My approach is incorrect. Sorry, for suggesting you wrong approach.

Could you find a test case for the same?

I think that it might just hold true in this case, just a possibility.

I tried many small random cases but the solution is giving correct answer actually. I am also surprised to see that the solution is working. Actually, the above code without memoization is surely correct but I thought when we would apply memoization then it would become incorrect because of the same reason you provided above. Currently, I am also not completely satisfied regarding why the solution is working. So, if you get the intuition behind the correctness please do share.

It fails. We can write inequalities and show.

So, did you get the correct solution with the complexity that we want to achieve???

No :(

Can you please tell me the name of the book?

It came in my exam!

Sorry,I didnt see the 7 marks in the brackets.

Sort all the values in increasing order of a_i's. We can then do something like dp[i][j][level] where the dp value denotes the cost for the binary search tree on values from a_i to a_j, with the root at depth = level. The recurrence is dp[i][j][level] = min(dp[i][k-1][level+1] + dp[k+1][j][level]+w_k*level^2) for k from i to j. This is O(N^2) space (we can keep only 2 columns for the level), but O(N^4) time.

I wonder if it's possible in with D&C optimization on each layer of level... not sure but I'll try to prove it.

I will continue from the solutions proposed by others. Let

dp_{h, l, r}be the best value we can get if we arrange the numbers indexed fromltorsuch that the root of the tree is at depthh. The recurrence is, of course:dp_{h, l, r}=min_{l ≤ i ≤ r}(dp_{h + 1, l, i - 1}+w_{i}h^{2}+dp_{h + 1, i + 1, r})We can build the solution in layers. The last layer is

h=n, and here we only need to computedp_{h, l, l}, since we can never build a tree with more than one node at depthn. The same logic applies for other values ofh. For example, whenh=n- 1, we only need to compute DP values wherer-l≤ 1. In general, we only need to compute DP values wherer-l≤n-h.I think that Knuth's optimization trick will work here. This would mean that we can compute each DP state in amortized constant time, so

O(n^{3}) total. Since we only need to track the last two DP layers this isO(n^{2}) memory.Can anyone prove the required property or give a counterexample?

EDIT:

Here's my stress test (100 random samples), and the

O(n^{3}) solution passed.