**K Stack Pops - Google Top Interview Questions**

### Problem Statement :

You are given two-dimensional list of integers stacks and an integer k. Assuming each list in stacks represents a stack, return the maximum possible sum that can be achieved from popping off exactly k elements from any combination of the stacks. Constraints n ≤ 500 where n is the number of rows in stacks. m ≤ 200 where m is the maximum number of elements in a stack. k ≤ 100 Example 1 Input stacks = [ [100, -3, -10], [1], [4, 5, 6] ] k = 4 Output 93 Explanation We can pop off all 3 elements from the first stack and pop the last element of the last stack to get -10 + -3 + 100 + 6. Example 2 Input stacks = [ [1, 2, 4], [2, 3, 5], [3, 0, 6] ] k = 3 Output 15 Explanation Pop off the 4, 5 and 6 from each of the stacks.

### Solution :

` ````
Solution in C++ :
int solve(vector<vector<int>>& rows, int k) {
vector<int> dp(k + 1, -1e9);
dp[0] = 0;
for (const auto& nums : rows) {
int n = nums.size();
auto dpc = dp;
for (int i = 1; i <= k; ++i) {
int sum = 0;
for (int j = 1; j <= min(i, n); ++j)
dpc[i] = max(dpc[i], (sum += nums[n - j]) + dp[i - j]);
}
dp = dpc;
}
return dp[k];
}
```

` ````
Solution in Java :
import java.util.*;
class Solution {
Integer[][] dp;
List<List<Integer>> total;
public int solve(int[][] stacks, int k) {
dp = new Integer[stacks.length][k + 1];
total = new ArrayList();
for (int i = 0; i < stacks.length; i++) {
List<Integer> list = new ArrayList();
list.add(0);
for (int j = stacks[i].length - 1; j > -1 && j >= stacks[i].length - k; j--) {
list.add(stacks[i][j] + list.get(list.size() - 1));
}
total.add(list);
}
return recur(0, k);
}
public int recur(int ind, int k) {
if (k == 0)
return 0;
if (ind == dp.length || k < 0)
return -1 * (int) (1e7);
if (dp[ind][k] != null)
return dp[ind][k];
int a = -1 * (int) (1e7);
for (int i = 0; i < total.get(ind).size(); i++) {
a = Math.max(a, total.get(ind).get(i) + recur(ind + 1, k - i));
}
dp[ind][k] = a;
return a;
}
}
```

` ````
Solution in Python :
class Solution:
def solve(self, stacks, k):
NINF = float("-inf")
dp = [NINF] * (k + 1)
dp[0] = 0
for stack in stacks:
P = [0]
for x in reversed(stack):
P.append(P[-1] + x)
for j in range(k, 0, -1):
for i in range(1, min(j + 1, len(P))):
dp[j] = max(dp[j], dp[j - i] + P[i])
return dp[k]
```

## View More Similar Problems

## Array-DS

An array is a type of data structure that stores elements of the same type in a contiguous block of memory. In an array, A, of size N, each memory location has some unique index, i (where 0<=i<N), that can be referenced as A[i] or Ai. Reverse an array of integers. Note: If you've already solved our C++ domain's Arrays Introduction challenge, you may want to skip this. Example: A=[1,2,3

View Solution →## 2D Array-DS

Given a 6*6 2D Array, arr: 1 1 1 0 0 0 0 1 0 0 0 0 1 1 1 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 An hourglass in A is a subset of values with indices falling in this pattern in arr's graphical representation: a b c d e f g There are 16 hourglasses in arr. An hourglass sum is the sum of an hourglass' values. Calculate the hourglass sum for every hourglass in arr, then print t

View Solution →## Dynamic Array

Create a list, seqList, of n empty sequences, where each sequence is indexed from 0 to n-1. The elements within each of the n sequences also use 0-indexing. Create an integer, lastAnswer, and initialize it to 0. There are 2 types of queries that can be performed on the list of sequences: 1. Query: 1 x y a. Find the sequence, seq, at index ((x xor lastAnswer)%n) in seqList.

View Solution →## Left Rotation

A left rotation operation on an array of size n shifts each of the array's elements 1 unit to the left. Given an integer, d, rotate the array that many steps left and return the result. Example: d=2 arr=[1,2,3,4,5] After 2 rotations, arr'=[3,4,5,1,2]. Function Description: Complete the rotateLeft function in the editor below. rotateLeft has the following parameters: 1. int d

View Solution →## Sparse Arrays

There is a collection of input strings and a collection of query strings. For each query string, determine how many times it occurs in the list of input strings. Return an array of the results. Example: strings=['ab', 'ab', 'abc'] queries=['ab', 'abc', 'bc'] There are instances of 'ab', 1 of 'abc' and 0 of 'bc'. For each query, add an element to the return array, results=[2,1,0]. Fun

View Solution →## Array Manipulation

Starting with a 1-indexed array of zeros and a list of operations, for each operation add a value to each of the array element between two given indices, inclusive. Once all operations have been performed, return the maximum value in the array. Example: n=10 queries=[[1,5,3], [4,8,7], [6,9,1]] Queries are interpreted as follows: a b k 1 5 3 4 8 7 6 9 1 Add the valu

View Solution →