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 天平