Best Time To Buy and Sell Stock
Posted: 7 Jan, 2021
You have been given an array 'PRICES' consisting of 'N' integers where PRICES[i] denotes the price of a given stock on the i-th day. You are also given an integer 'K' denoting the number of possible transactions you can make.
Your task is to find the maximum profit in at most K transactions. A valid transaction involves buying a stock and then selling it.
You can’t engage in multiple transactions simultaneously, i.e. you must sell the stock before rebuying it.
Input: N = 6 , PRICES = [3, 2, 6, 5, 0, 3] and K = 2. Output: 7 Explanation : The optimal way to get maximum profit is to buy the stock on day 2(price = 2) and sell it on day 3(price = 6) and rebuy it on day 5(price = 0) and sell it on day 6(price = 3). The maximum profit will be (6 - 2) + (3 - 0) = 7.
The first line of input contains an integer 'T' which denotes the number of test cases or queries to be run. Then the test cases follow. The first line of each test case contains two single-space separated integers ‘N’ and ‘K’, respectively. The second line of each test case contains ‘N’ single space-separated integers, denoting the elements of the array 'PRICES'.
For each test case, print a single integer denoting the maximum profit in at most K transactions. Print the output of each test case in a separate line.
You do not need to print anything; it has already been taken care of. Just implement the given function.
1 <= T <= 100 1 <= N <= 5000 0 <= K <= N/2 0 <= ARR[i] <=10^5 Time Limit : 1 sec
The main idea is to use recursion to reduce the big problem into several smaller subproblems.
- We will call a maxProfit function that returns us the maximum profit we can get from the array starting from i (here i denotes the starting index of PRICES) to N - 1. We will use a bool variable buy. If buy == true it denotes that we had bought a stock and now we can’t buy another until we sell it and buy == false denotes that we hadn’t currently bought any stock. The maxProfit function will work as follows:
- If K == 0 || i == N
- Return 0.
- If buy == false
- We make a recursive call to i + 1 with buy marked as false. (Not holding the stock).
- We make a recursive call to i + 1 with buy marked as true and subtract PRICES[i]. (Buy stock on i -th day).
- We finally return the maximum of them.
- We make a recursive call to i + 1 with buy marked as true. (Hold the stock).
- We make a recursive call to i + 1 with buy marked as false and add PRICES[i] and decrement K by 1. (Sell stock on i-th day).
- Return the maximum of them.