papamelon 201. 部分和问题

地址 https://www.papamelon.com/problem/201

解答
使用dfs遍历各个数字选择或者不选择的情况,判断是否能够得到指定的和
时间复杂度 O(2^n)

#define _CRT_SECURE_NO_WARNINGS
#include <iostream>


using namespace std;

const int N = 50;
int arr[N];
int n, k,sum;


int dfs(int idx) {
	if (sum == k) { 
		return 1; 
	}
	if (idx >= n) { return 0; }

	//尝试不选择该数
	if (1 == dfs(idx + 1)) { return 1; }

	//尝试选择该数
	sum += arr[idx]; idx++;
	if (1 == dfs(idx)) { return 1; }
	idx--; sum -= arr[idx];

	return 0;
}


int main()
{
	while (~scanf("%d", &n)) {
		sum = 0; 
		for (int i = 0; i < n; i++) { scanf("%d",&arr[i]); }
		scanf("%d",&k);
		if (1 == dfs(0)) {
			printf("Yes\n");
		}
		else {
			printf("No\n");
		}
	}

	

	return 0;
}

posted on 2021-11-12 11:25  itdef  阅读(59)  评论(0编辑  收藏  举报

导航