poj1909 Marbles on a tree

来源:互联网 发布:ubuntu安装libcurl 编辑:程序博客网 时间:2024/06/17 22:01
一看就会
#include <iostream>#include <cstdio>#include <cmath>#include <cstring>#include <vector>using namespace std;const int M = 10005;vector<int>que[M];int num[M];int rd[M];int n, ans;void init() {    for(int i = 0; i <= n; i++){        que[i].clear();        num[i] = 0;        rd[i] = 0;    }}void dfs(int u){     int v;    for(int i = 0; i < (int)que[u].size(); i++) {        v = que[u][i];        dfs(v);        num[u] += num[v];        ans += fabs(num[v]);    }    num[u] -= 1;}int main(){    int a, b, q;    while(scanf("%d", &n) != EOF) {        if(!n)            break;         init();         ans = 0;        for(int i = 0; i< n; i++){            scanf("%d%d%d", &a, &b, &q);            num[a] = b;            for(int j = 0; j < q; j++){                scanf("%d", &b);                que[a].push_back(b);                rd[b]++;            }        }        for(int i = 1; i <= n; i++){            if(!rd[i]){                dfs(i);                break;            }        }        printf("%d\n", ans);    }    return 0;}

0 0
原创粉丝点击