P2415 集合求和

CodeSlogan

已于 2022-03-14 17:31:32 修改

阅读量921

点赞数 1

分类专栏：算法题文章标签：深度优先 c++

于 2022-03-14 17:30:52 首次发布

本文链接：https://blog.csdn.net/comscience/article/details/123483807

版权

算法题专栏收录该内容

70 篇文章 1 订阅

订阅专栏

题目描述

给定一个集合s（集合元素数量<=30），求出此集合所有子集元素之和。

输入格式

集合中的元素（元素<=1000）

输出格式

和

输入输出样例

输入 #1

2 3

输出 #1

说明/提示

子集为：

[] [2] [3] [2 3] 2+3+2+3=10

保证结果在10^18以内。

60%

这道题本意是考察数学规律，但是我当时想成了dfs，而且还没写出来，会超时，只拿到3个测试点的分，就挺遗憾。
在这里插入图片描述

#include <iostream>
#include <cstring>
#include <cmath>
using namespace std;
typedef long long LL;
const int N = 40;
int a[N];
LL ans = 0;
int idx = 1;
bool st[N];

void dfs(int u) {
	if (u == idx) {
		for (int i = 1; i < idx; i++)
			if (st[i]) {
				ans += a[i];
			}
		return;
	}
	
	st[u] = true;
	dfs(u+1);
	st[u] = false;
	dfs(u+1);
}

int main() {
	int n;
	while(cin >> n) {
		a[idx] = n;
		idx++;
	}
	dfs(1);
	cout << ans << endl;
	
	return 0;
}