首页 > 代码库 > 搜索场 day1A 天平
搜索场 day1A 天平
A 天平
时间限制: 1 Sec 空间限制: 256 MB
输入输出文件名:A.in,A.out
题目描述
给出N(1≤N≤20)个整数(数字在1...100,000,000之间)在其中选若干个数,如果这几个数可以分成两个和相等的集合,那么方案数加1。问总方案数。
输入格式:
第一行为n,第二行为n个数字。
输出格式:
输出方案数
输入样例
4
1 2 3 4
输出样例
3
样例解释
{1,2,3},{1,3,4},{1,2,3,4}三种方案。
数据范围
存在部分数据点n<=10
对于100%的数据,1<=n<=20
#include<cstdio> #include<cstring> #define ll long long using namespace std; const int N=100010; struct node { int next,to,p; }e[N*2]; int first[N],book[N],size[N]; int cnt=0; ll sum=0; int n; void insert(int v,int u,int p) { e[++cnt].to=v;e[cnt].next=first[u];first[u]=cnt;e[cnt].p=p; e[++cnt].to=u;e[cnt].next=first[v];first[v]=cnt;e[cnt].p=p; } void dfs1(int x) { book[x]=1;size[x]=1; for(int i=first[x];i;i=e[i].next) { if(book[e[i].to]==0) { dfs1(e[i].to); size[x]+=size[e[i].to]; } } } void dfs(int x) { book[x]=1; for(int i=first[x];i;i=e[i].next) { if(book[e[i].to]==0) { sum+=(ll)e[i].p*size[e[i].to]*(n-size[e[i].to]); dfs(e[i].to); } } } int main() { freopen("A.in","r",stdin); freopen("A.out","w",stdout); scanf("%d",&n); int v,u,p; for(int i=1;i<=n-1;i++) { scanf("%d %d %d",&u,&v,&p); insert(u,v,p); } memset(book,0,sizeof(book)); dfs1(1); memset(book,0,sizeof(book)); dfs(1); printf("%lld\n",sum); }
谢谢师兄(czl)教导。
搜索场 day1A 天平
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。