联合权值

联合权值

https://ac.nowcoder.com/acm/problem/16495

做法:DFS,树的深度优先遍历

思路:

距离为2的点对有两种:
图片说明
1.自己的儿子们之间
2.自己的父亲和自己的儿子们

对树进行一遍dfs后得出max和sum
因为是求有序点对(dfs是从上到下,然而有序对也可以是从下往上),所以结果需要*2

代码

#include <bits/stdc++.h>
using namespace std;
#define pb push_back
#define mp(aa,bb) make_pair(aa,bb)
#define _for(i,b) for(int i=(0);i<(b);i++)
#define rep(i,a,b) for(int i=(a);i<=(b);i++)
#define per(i,b,a) for(int i=(b);i>=(a);i--)
#define mst(abc,bca) memset(abc,bca,sizeof abc)
#define X first
#define Y second
#define lowbit(a) (a&(-a))
typedef long long ll;
typedef pair<int,int> pii;
typedef unsigned long long ull;
typedef long double ld;
const int N=2e5+10;
const int INF=0x3f3f3f3f;
const int mod=10007;
const double eps=1e-6;
const double PI=acos(-1.0); 

int n,w[N],Max,Sum;
vector<int> g[N];

void dfs(int u,int fa){
    int sumson=0,maxson=0;
    for(int v:g[u]){
        if(v==fa) continue;
        dfs(v,u);
        //自己(u)的儿子们(v) 
        Max=max(Max,w[v]*maxson);
        Sum+=w[v]*sumson;
        Sum%=mod;

        sumson+=w[v];
        sumson%=mod;
        maxson=max(maxson,w[v]);                                                                            
    }
    //自己(u)父亲(fa)和自己儿子们(v)    
    Max=max(Max,w[fa]*maxson);
    Sum+=w[fa]*sumson;
    Sum%=mod;
}

void solve(){
    cin>>n;
    _for(i,n-1){
        int u,v;cin>>u>>v;
        g[u].pb(v);g[v].pb(u);
    }
    rep(i,1,n) cin>>w[i];
    dfs(1,0);
    cout<<Max<<" "<<Sum*2%mod<<"\n";
}

int main(){
    ios::sync_with_stdio(0);
    cin.tie(0);
#ifdef DEBUG
    freopen("F:/laji/1.in", "r", stdin);
//    freopen("F:/laji/2.out", "w", stdout);
#endif
//     int t;cin>>t;while(t--)
    solve();
    return 0;
}
牛客每日一题 文章被收录于专栏

全部评论

相关推荐

03-06 12:44
已编辑
吉林大学 Java
是个千人厂,没听过名字。1.&nbsp;做一个自我介绍。2.&nbsp;你这个项目和技术栈从哪里学的?有报辅导班嘛[答&nbsp;都是是自己网上学的,学校教的东西没用]3.&nbsp;我看了你放在github上的项目,前端也是你写的嘛[答&nbsp;AI写的,90%精力用于后端开发,前端单纯用于作为后端逻辑的可视化技术验证(骗你的其实后端也是AI写的)]4.&nbsp;好,你觉得这些技术栈研究得最深刻的是哪个[答&nbsp;八股压根没背到后面,昨晚背了MySQL就说MySQL]5.&nbsp;那讲一下MySQL的索引[答&nbsp;从B+树选型一路吟唱到联合索引,索引失效]6.&nbsp;联合索引ABC问题,AB走索引嘛,BC走索引嘛?BAC走索引嘛?A&nbsp;or&nbsp;B&nbsp;走索引嘛[走,不走,走,不走。面试官点头说可以]7.&nbsp;讲一下项目里Redission分布式锁实现8.&nbsp;Watchdog机制具体是怎么工作9.&nbsp;消息队列有考虑过Kafka嘛,怎么选型的10.&nbsp;你这个项目消息队列可能出现什么问题,怎么解决这个问题?[瞎扯没用的,被面试官引导答了视频处理可能产生消息堆积问题,然后开始吟唱]11.&nbsp;文件分片自己写的还是用的什么框架?上传进度的Redis数据结构?上传的视频有多大?小分片大小?12.&nbsp;项目里Redis会话记忆是啥意思?[面试官说不行,没人把这个全放Redis里[生气R]]13.&nbsp;那这和直接查数据库有什么区别[扯了Token成本和解决幻觉问题之类的,给面试官听笑了,我最后也没绷住]14.&nbsp;你平时是怎么使用AI&nbsp;coding的15.&nbsp;算法,给了我一个leedcode链接,一看做过了。然后换了一道三数之和,也做过了。然后面试官说算了,让我讲讲思路吧反问:1.有什么需要提高的地方2.介绍一下部门业务有哪些这个面试官真的感官非常非常好,问问题还疯狂引导,感觉不会也会了。找实习&nbsp;&nbsp;牛客AI配图神器#
查看15道真题和解析
点赞 评论 收藏
分享
要AC不要WA:投一天,喜提两笔试
点赞 评论 收藏
分享
评论
1
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务