HDU 1520 Anniversary party(단순 트리 dp)

1404 단어 트리 dp
방금 비슷한 걸 했는데 계속wa, 왜, 인터넷 검색 문제가 하나 더 풀렸어 n!0의 판단은 ac입니다. 그런데 제목에서 말하지 않은 것 같은데...
사고방식은 수조를 이용하여 뿌리 노드를 찾은 후에 dfs가 차례로 돌아가는 것이다
HDU - 1520
#include
#include
#include
#include
#include
#include
#include
#include
#include
using namespace std;
int dp[6005][2];
vector v[6005];
int vis[6005];
void dfs(int cur)
{
    for(int i = 0; i < v[cur].size(); i ++)
    {
        int son = v[cur][i];
        dfs(son);
        dp[cur][1] += dp[son][0];
        dp[cur][0] += max(dp[son][0], dp[son][1]);
    }
}
int main()
{
    int n;
    while(scanf("%d",&n) != EOF && n)
    {
        memset(dp, 0, sizeof(dp));
        memset(vis, 0, sizeof(vis));
        for(int i = 1; i <= n; i ++){
            scanf("%d",&dp[i][1]);
            v[i].clear();
        }
        int u, vv;
        while(scanf("%d%d",&u,&vv) && (u + vv))
        {
            v[vv].push_back(u);
            vis[u] = vv;
        }
        int ans;
        for(int i = 1; i <= n; i ++)
            if(! vis[i]){
                dfs(i);
                ans = i;
                break;
        }
        printf("%d
",max(dp[ans][0], dp[ans][1])); } return 0; }

좋은 웹페이지 즐겨찾기