P1351 联合权值
生活随笔
收集整理的這篇文章主要介紹了
P1351 联合权值
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
為了寫一寫LCA,我就按照標簽找……結果這道題我寫完竟然沒用LCA……真是神奇。。。
很多人(包括我),首先就想到了要枚舉每一個點,再枚舉任意這個點的兩個兒子,可是顯然O(n2)會T……
其實我們只要線性掃一遍就可以了,利用小學學到的乘法分配率,邊走邊加val,這樣下一個點和val的乘積就是它和這之前所有的點的乘積之和,最后就是sum。
當然,點對反過來就是一組新的點對,所以,隨后的sum要乘2。
而對于最大值,只要每一次更新我掃過的這些點權里最大的數是多少(在一次外層循環中),和新的點權相乘,看看能不能更新最大解即可。
對了,記得看看是給誰取模。
代碼如下:
#include<cstdio> #include<iostream> #include<algorithm> #include<cstring> using namespace std; #define maxn 4000000 #define mod 10007 #define int long long int head[maxn],to[maxn],nxt[maxn],w[maxn]; int n,cnt; void add(int a,int b) {to[++cnt]=b;nxt[cnt]=head[a];head[a]=cnt; } main() {scanf("%lld",&n);for(int i=1;i<=n-1;i++){int a,b;scanf("%lld%lld",&a,&b);add(a,b);add(b,a);} for(int i=1;i<=n;i++)scanf("%lld",&w[i]);int sum=0,Max=0;int fir,val,maxpo;for(int i=1;i<=n;i++){fir=head[i];val=w[to[fir]]%mod;maxpo=w[to[fir]];fir=nxt[fir];for(; fir; fir=nxt[fir]){sum=(sum+val*w[to[fir]])%mod;val=(val+w[to[fir]])%mod;Max=max(Max,maxpo*w[to[fir]]);maxpo=max(maxpo,w[to[fir]]);}}printf("%lld %lld",Max,(2*sum)%mod);return 0; }?
轉載于:https://www.cnblogs.com/popo-black-cat/p/10301580.html
總結
以上是生活随笔為你收集整理的P1351 联合权值的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 协程:Greenlet模块、Gevent
- 下一篇: java基础-2