dfs 70pts超时求调
查看原帖
dfs 70pts超时求调
1036776
zjy4楼主2025/7/22 09:31

代码放这里了

#include<bits/stdc++.h>
using namespace std;
#define int long long 
int n,x,y,a[400001],ans=0,maxn=0,vis[400001];
vector<int> e[400001];
void dfs(int x){
    if(vis[x]==1){
        for(int i:e[x]){
            for(int u:e[i]){
                if(u!=x) {
                    ans+=a[u]*a[x];
                    ans%=10007;
                    maxn=max(a[u]*a[x],maxn);
                }
            }
        }
        vis[x]=2;
        for(int i:e[x]){
            if(vis[i]!=2) dfs(i)
        };//在样例里不执行这段dfs就无法处理5号
        return;
    }
    vis[x]=1;
    for(int i:e[x]){
        if(vis[i]!=2) dfs(i);
    }
}
signed main(){
    cin>>n;
    for(int i=1;i<n;i++){
        cin>>x>>y;
        e[x].push_back(y);
        e[y].push_back(x);
    }
    for(int i=1;i<=n;i++) cin>>a[i];
    dfs(1);
    cout<<maxn<<" "<<ans%10007;
}

数组已经开了2倍了

2025/7/22 09:31
加载中...