Участник:Михеева Анастасия Максимовна/Max Sum of Rectangle No Larger Than K

Материал из DISCOPAL
Перейти к: навигация, поиск

class Solution {

   public:   
   int maxSumSubmatrix(vector<vector<int>> &matrix, int k) {
      int row = matrix.size();
      if (row == 0)
          return 0;
      int col = matrix[0].size();
      int ret = INT_MIN;
      for (int l = 0; l < col; l++) {
         vector<int> sums(row, 0);
         for (int r = l; r < col; r++) {
            for (int i = 0; i < row; i++)
               sums[i] += matrix[i][r];
            set<int> sumSet;
            sumSet.insert(0);
            int curSum = 0;
            int curMax = INT_MIN;
            for (auto sum : sums) {
               curSum += sum;
               auto it = sumSet.lower_bound(curSum — k);
               if (it != sumSet.end())
                 curMax = max(curMax, curSum — *it);
               sumSet.insert(curSum);
            }
            ret = max(ret, curMax);

}

       }

return ret; }};