如果存在具有和K的子阵列

时间:2016-09-18 05:31:33

标签: c++ arrays algorithm performance

给定一个正整数数组,这是查找是否存在子数组(不一定是连续的)的最有效方法,使得其元素之和等于给定数字k。登记/> 我的方法是一个天真的递归:

#include <iostream>
#include <algorithm>
#include <utility>
#include <stdio.h>
#include <queue>
#include <climits>
#include <malloc.h>
#include <stdlib.h>
#include <cstring>
#include <cmath>
#include <sstream>
#include <map>
#include <stack>
#include <string>
#include <ctime>
using namespace std;
int flag=0;
void solve(int ind,int n,int arr[],int sum,int k)
{
    if(sum>k)
        return;
    if(sum==k){
        flag++;
        return;
    }
    for(int i=ind+1;i<n;i++)
        solve(i,n,arr,sum+arr[i],k);
}
int main()
{
    int t;
    cin >> t;//number of test cases
    while(t--)
    {
        int n,k;//n->number of array elements
        cin >> n >> k;//k->given number
        int arr[n];
        for(int i=0;i<n;i++)
            cin >> arr[i];
        for(int i=0;i<n;i++)
            solve(i,n,arr,arr[i],k);
        if(flag>0)
            cout << "YES!" << endl;
        else
            cout << "NO!" << endl;
        flag=0;
    }
    return 0;
} 

0 个答案:

没有答案