ICPC Nanjing 2021 H, Crystalfly

最新推荐文章于 2024-10-12 22:09:48 发布

原创最新推荐文章于 2024-10-12 22:09:48 发布 · 487 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#深度优先 #算法

dp 专栏收录该内容

7 篇文章

订阅专栏

本文介绍了一种使用深度优先搜索（DFS）在给定图中寻找最优路径并计算节点价值的问题。通过递归实现，针对特定条件（如节点值为3时的选择策略），优化了路径选择，最终求解得到每个节点的最优路径价值。

#include<bits/stdc++.h>
using namespace std;
#define ll long long
inline char gc()
{
    static char buf[1000000],*p1=buf,*p2=buf;
    return p1==p2&&(p2=(p1=buf)+fread(buf,1,1000000,stdin),p1==p2)?EOF:*p1++;
}
#define gc getchar
inline ll read()
{
    char c=gc();
    ll su=0,f=1;
    for (; c<'0'||c>'9'; c=gc()) if (c=='`-') f=-1;
    for (; c>='0'&&c<='9'; c=gc()) su=su*10+c-'0';
    return su*f;
}
inline void write(ll x)
{
    if (x<0)
    {
        putchar('-');
        write(-x);
        return;
    }
    if (x>=10) write(x/10);
    putchar(x%10+'0');
}
inline void writesp(ll x)
{
    write(x),putchar(' ');
}
inline void writeln(ll x)
{
    write(x);
    putchar('\n');
}
constexpr int maxn=1e5+10;
constexpr ll mod=998244353;
#define bug(x) cout<<#x<<"=="<<x<<endl;
#define fp(i,a,b) for(register int i=a,I=b+1;i<I;++i)
#define fd(i,a,b) for(register int i=a,I=b-1;i>I;--i)
#define P pair<int,ll>
ll a[maxn]= {0};
int t[maxn]= {0};
vector<int> way[maxn];
int main()
{
    int T=read();
    while(T--)
    {
        int n=read();
        vector<ll> a(n+1);
        vector<int> t(n+1);
        for(int i=1; i<=n; ++i)
        {
            a[i]=read();
        }
        for(int i=1; i<=n; ++i)
        {
            t[i]=read();
        }
        vector< vector<int> > way(n+1);
        for(int i=1; i<=n-1; i++)
        {
            int u=read();
            int v=read();
            way[u].push_back(v);
            way[v].push_back(u);
        }
        vector<ll> g(n+1);
        vector<ll> h(n+1);
        function<void(int,int)> dfs= [&](int u,int fu){
            g[u]=0;///不选择u点
            h[u]=a[u];///选择该u点
            ll maxx1=0;
            ll maxx2=0;
            ll sum=0;
            for(int v:way[u]){
                if(v==fu)continue;
                dfs(v,u);
                h[u]+=g[v];///不选择v点的g[v]值是一定可以取得
                if(t[v]==3){///为3的时候是a[v]点一定可以取的状态,所以选两个最大和次大即可
                    if(a[v]>maxx1){
                        maxx2=maxx1;
                        maxx1=a[v];
                    }else if(a[v]>maxx2){
                        maxx2=a[v];
                    }
                }
                sum+=g[v];
            }
            ll ans=0;
            for(int v:way[u]){
                if(v==fu)continue;
                ans = max(ans, h[v] + (sum - g[v]) + ((a[v] == maxx1)? maxx2: maxx1));
                ///选取一个点用来占v，选取一个点拥有次大或者最大值
                ans=max(ans,sum+a[v]);///选取一个最大的点来取
            }
            g[u]=ans;
        };
        dfs(1,0);
        writeln(g[1]+a[1]);
    }
}