神刀安全网

Number of ways to calculate a target number using only array elements

Given an integer array, find number of ways to calculate a target number using only array elements and addition or subtraction operator.

Example: Input: arr[] = {-3, 1, 3, 5}, k = 6 Output: 4  Explanation -  - (-3) + (3) + ( 1) + (5) + (-3) + (1) + (3) + (5) - (-3) + (1) - (3) + (5)

The problem is similar to0-1 Knapsack Problem where for every item, we either pick the complete item, or don’t pick it at all (0-1 property). The idea remains the same here i.e. we either include the current digit or ignore it. If we include the current digit, we subtract or add it from remaining target and recurse for remaining digits with new target. If target reaches 0, we increment the count. If we have processed all elements of the array and target is not reached, count remains unchanged.

Below is recursive C++ implementation of above idea.

// C++ program to find the number of ways to calculate // a target number using only array elements and // addition or subtraction operator. #include <iostream> #include <vector> using namespace std;  // Function to find the number of ways to calculate // a target number using only array elements and // addition or subtraction operator. int findTotalWays(vector<int> arr, int i, int k) {     // If all elements are processed and     // target is not reached, return 0     if (i >= arr.size() && k != 0)         return 0;      // If target is reached, return 1     if (k == 0)         return 1;      // Return total count of three cases     // 1. Don't consider current element     // 2. Consider current element and subtract it from target     // 3. Consider current element and add it to target     return findTotalWays(arr, i + 1, k)            + findTotalWays(arr, i + 1, k - arr[i])            + findTotalWays(arr, i + 1, k + arr[i]); }  // Driver Program int main() {     vector<int> arr = {-3, 1, 3, 5, 7};      // target number     int k = 6;      cout << findTotalWays(arr, 0, k) << endl;      return 0; }

Output :

Time Complexity of above solution is O(3^n) i.e. exponential.

This article is contributed by Aditya Goel . If you like GeeksforGeeks and would like to contribute, you can also write an article and mail your article to contribute@geeksforgeeks.org. See your article appearing on the GeeksforGeeks main page and help other Geeks.

Please write comments if you find anything incorrect, or you want to share more information about the topic discussed above

转载本站任何文章请注明:转载至神刀安全网,谢谢神刀安全网 » Number of ways to calculate a target number using only array elements

分享到:更多 ()

评论 抢沙发

  • 昵称 (必填)
  • 邮箱 (必填)
  • 网址