ZOJ 2374 Marbles on a tree

查看原题

题意

把每一个节点上的大理石均匀移到树的每一个节点上，一次只能向相邻的移一个，问最少需要多少步骤

思路

从子节点算起，大于1的就把1以外的部分向上传，小于1的就往上拿1
注意要建立一个队列保存没有子节点的节点

代码

#include <iostream>
#include <queue>
using namespace std;
struct node{
    int marbles,fathernode,childnum;
}nodes[10010];
int main(){
    int n;
    while(cin>>n){
        if(n==0){
            return 0;
        }
        queue<int>que;
        int moves=0;
        for(int i=1;i<=n;i++){
            int childs,marble,number;
            cin>>number>>marble>>childs;
            nodes[number].marbles=marble;
            nodes[number].childnum=childs;
            if(childs==0)que.push(number);
            while(childs--){
                int temp;
                cin>>temp;
                nodes[temp].fathernode=number;
            }
        }
        while(!que.empty()){
            int i=que.front();
            que.pop();
            nodes[nodes[i].fathernode].childnum--;
            if(nodes[nodes[i].fathernode].childnum==0)que.push(nodes[i].fathernode);
            if(nodes[i].marbles>1){
                int temp=nodes[i].marbles-1;
                nodes[i].marbles=1;
                nodes[nodes[i].fathernode].marbles+=temp;
                moves+=temp;
            }
            if(nodes[i].marbles<1){
                int temp=1-nodes[i].marbles;
                nodes[i].marbles=1;
                nodes[nodes[i].fathernode].marbles-=temp;
                moves+=temp;
            }
        }
        cout<<moves<<endl;
    }
    return 0;
}