hiho一下 第109周:Tower Defense Game

题目链接

参考了下别人的思路（推荐一篇博客讲解的很清楚为什么那样排序：点击打开链接），所以是按q从大到小排序的顺序来决定先经过哪个节点，还有就是给出的两点之间有边的应该2个点都保存相应的连接点。

用dfs的思想，每次先更新完子节点再更新当前节点。我又传进了当前节点的父节点，因为连接关系中没保存父子关系，所以要把邻接点中自己的父节点排除掉，只更新子节点。

之前错在了更新一个点的p,q值上，正确思路是确定顺序后，保存当前需要的最大钱数和剩余钱数，初始化为当前父节点的p,q，然后每次经过一个节点就判断当前的钱是否足够付当前点的支出，如果不够就要补上，这时p值就要增加相应的钱说明带的钱不够。q也就变为当前点的挣的钱。如果够付支出，就更新q值为付完后剩下的钱加上挣的钱。

#include <iostream>
#include<vector>
#include<algorithm>
using namespace std;
 vector<int> link[10010],v1;
struct node{
      int p,q;
}nodes[10010];//保存n个点的支出和收入
bool cmp(int n1,int n2){
      return nodes[n1].q>nodes[n2].q;//按q从大到小排
}
void dfs(int root,int fa){
    int p1=nodes[root].p;//p1记录需要付出的最大价钱
    int q1=nodes[root].q;//q1记录剩余的钱
    vector<int >v1=link[root];
    for(int i=0;i<v1.size() ;i++){
       if(v1[i]!=fa){
            dfs(v1[i],root);//初始化子节点，除了父节点
       }
    }
    sort(v1.begin(),v1.end(),cmp);//对子节点按q从大到小排序
    for(int i=0;i<v1.size() ;i++){
       if(v1[i]!=fa){
        if(q1<nodes[v1[i]].p){
          p1+=nodes[v1[i]].p-q1;//若剩余的钱不够就要不足缺少的部分
          q1=nodes[v1[i]].q;//剩余的前就变为当前节点的q
        }
        else{
            q1=q1-nodes[v1[i]].p+nodes[v1[i]].q;//若若剩余的钱够就更新剩余的钱为减去支出的加上挣来的。
        }
       }
    }
    nodes[root].p=p1;
    nodes[root].q=q1;
}
int main()
{
    int n,a,b;
    cin>>n;
    for(int i=1;i<=n;i++)
        cin>>nodes[i].p>>nodes[i].q;
    for(int i=1;i<n;i++){
        cin>>a>>b;
       // if(a<b)
        link[a].push_back(b);//保存有连接的点
       // else
        link[b].push_back(a);
    }
     dfs(1,0);

    cout << nodes[1].p << endl;
    return 0;
}