diff --git a/backtracking/subset_sum.cpp b/backtracking/subset_sum.cpp index b9dee734b..2428e8ab8 100644 --- a/backtracking/subset_sum.cpp +++ b/backtracking/subset_sum.cpp @@ -31,13 +31,13 @@ namespace subset_sum { * @returns count of the number of subsets with required sum */ -uint64_t subset_sum(int sum, const std::vector &in_arr) { - int nelement = in_arr.size(); - int count_of_subset = 0; +uint64_t subset_sum(int32_t sum, const std::vector &in_arr) { + int32_t nelement = in_arr.size(); + uint64_t count_of_subset = 0; - for (int i = 0; i < (1 << (nelement)); i++) { - int check = 0; - for (int j = 0; j < nelement; j++) { + for (int32_t i = 0; i < (1 << (nelement)); i++) { + int32_t check = 0; + for (int32_t j = 0; j < nelement; j++) { if (i & (1 << j)) { check += (in_arr[j]); } @@ -58,7 +58,7 @@ uint64_t subset_sum(int sum, const std::vector &in_arr) { static void test() { // Test 1 std::cout << "1st test "; - std::vector array1 = {-7, -3, -2, 5, 8}; // input array + std::vector array1 = {-7, -3, -2, 5, 8}; // input array assert(backtracking::subset_sum::subset_sum(0, array1) == 2); // first argument in subset_sum function is the required sum and // second is the input array @@ -66,34 +66,28 @@ static void test() { // Test 2 std::cout << "2nd test "; - std::vector array2 = {1, 2, 3, 3}; - assert(backtracking::subset_sum::subset_sum(6, array2) == - 3); // here we are expecting 3 subsets which sum up to 6 i.e. + std::vector array2 = {1, 2, 3, 3}; + assert(backtracking::subset_sum::subset_sum(6, array2) == 3); // here we are expecting 3 subsets which sum up to 6 i.e. // {(1,2,3),(1,2,3),(3,3)} std::cout << "passed" << std::endl; // Test 3 std::cout << "3rd test "; - std::vector array3 = {1, 1, 1, 1}; - assert(backtracking::subset_sum::subset_sum(1, array3) == - 4); // here we are expecting 4 subsets which sum up to 1 i.e. + std::vector array3 = {1, 1, 1, 1}; + assert(backtracking::subset_sum::subset_sum(1, array3) == 4); // here we are expecting 4 subsets which sum up to 1 i.e. // {(1),(1),(1),(1)} std::cout << "passed" << std::endl; // Test 4 std::cout << "4th test "; - std::vector array4 = {3, 3, 3, 3}; - assert(backtracking::subset_sum::subset_sum(6, array4) == - 6); // here we are expecting 6 subsets which sum up to 6 i.e. - // {(3,3),(3,3),(3,3),(3,3),(3,3),(3,3)} + std::vector array4 = {3, 3, 3, 3}; + assert(backtracking::subset_sum::subset_sum(6, array4) == 6); // here we are expecting 6 subsets which sum up to 6 i.e. {(3,3),(3,3),(3,3),(3,3),(3,3),(3,3)} std::cout << "passed" << std::endl; // Test 5 std::cout << "5th test "; - std::vector array5 = {}; - assert(backtracking::subset_sum::subset_sum(6, array5) == - 0); // here we are expecting 0 subsets which sum up to 6 i.e. we - // cannot select anything from an empty array + std::vector array5 = {}; + assert(backtracking::subset_sum::subset_sum(6, array5) == 0); // here we are expecting 0 subsets which sum up to 6 i.e. we cannot select anything from an empty array std::cout << "passed" << std::endl; }