added requested changes in the code

This commit is contained in:
Pardeep Bhatt 2020-09-25 16:14:02 +05:30
parent c104eade81
commit 777bc0374d

View File

@ -1,35 +1,63 @@
/*Given a rod of length n inches and an array of prices that /**
contains prices of all pieces of size smaller than n. Determine * @file cut_rod.cpp
the maximum value obtainable by cutting up the rod and selling * @brief Implementation of cutting a rod problem
the pieces.*/ *
* @details
* Given a rod of length n inches and an array of prices that
* contains prices of all pieces of size smaller than n. Determine
* the maximum profit obtainable by cutting up the rod and selling
* the pieces.
*
*/
#include <array> #include <array>
#include <climits> #include <climits>
#include <iostream> #include <iostream>
using std::cout;
template <size_t T> template <size_t T>
int cutrod(const std::array<int, T> &p, const int n) { /**
int *r = new int[n + 1]; * This function cuts the rod in different pieces and stores the maximum profit
r[0] = 0; * for each piece of the rod.
for (int j = 0; j < n; j++) { * @param n size of the rod in inches
* @param price an array of prices that contains prices of all pieces of size<=n
* @return maximum profit obtainable for @param n inch rod.
*/
int cut_rod(const std::array<int, T> &price, const int n) {
// profit[i] will hold maximum profit for i inch rod
int *profit = new int[n + 1];
// if length of rod is zero, then no profit
profit[0] = 0;
// outer loop will select size of rod, starting from 1 inch to n inch rod.
// inner loop will evaluate the maximum profit we can get for i inch rod by
// making every possible cut on it and will store it in profit[i].
for (size_t i = 1; i <= n; i++) {
int q = INT_MIN; int q = INT_MIN;
for (int i = 0; i <= j; i++) { for (size_t j = 1; j <= i; j++) {
q = std::max(q, p[i] + r[j - i]); q = std::max(q, price[j - 1] + profit[i - j]);
} }
r[j + 1] = q; profit[i] = q;
} }
int ans = r[n]; int ans = profit[n];
delete[] r; delete[] profit;
return ans; return ans; /** return maximum profit obtainable for @param n inch rod */
} }
/**
* @brief Main function
* @returns 0 on exit
*/
int main() { int main() {
const int n = 30; const int n = 30; // size of rod
std::array<int, n> price = {1, 5, 8, 9, 10, 17, 17, 20, 24, 30, std::array<int, n> price = {
31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 1, 5, 8, 9, 10, 17, 17, 20, 24, 30, // price array
41, 42, 43, 44, 45, 46, 47, 48, 49, 50}; 31, 32, 33, 34, 35, 36, 37, 38, 39, 40,
cout << cutrod(price, n); 41, 42, 43, 44, 45, 46, 47, 48, 49, 50};
// maximum profit
std::cout << cut_rod(price, n);
return 0; return 0;
} }