タイトル説明
問題の解決策
思考
この質問は、暴力のそれはそれではないでしょうか?
Aソリューション
私は暴力的なことでしょう!
#include<iostream>
#include<vector>
using namespace std;
int n,ans,sum,a[200005],f[200005];
vector<int>g[200005];
void dfs(int u,int fa) {
for(int i=0; i<g[u].size(); i++) {
int x=g[u][i];
for(int j=0; j<g[u].size(); j++) {
int y=g[u][j];
if(x==y)continue;
ans=max(ans,a[x]*a[y]);
sum=(sum+a[x]*a[y])%10007;
}
}
for(int i=0; i<g[u].size(); i++) {
int v=g[u][i];
if(v==fa)continue;
dfs(v,u);
}
}
int main() {
cin>>n;
for(int i=1; i<n; i++) {
int u,v;
cin>>u>>v;
g[u].push_back(v);
g[v].push_back(u);
}
for(int i=1; i<=n; i++)cin>>a[i];
dfs(1,0);
cout<<ans<<" "<<sum;
return 0;
}
時間の複雑さ:\(O(^ N-2)\)
スコア期待:60ポイント
ソリューション2
考慮見出さ合成重みが同時にノードに接続された2つのノードで形成されています
このようなツリーの場合:
これは、の合計重量によるものである((1,3)、(2,4)、(4,5)\)\ 3点が無秩序構成します
私たちは、各ノードを通過する必要があり、それに接続されているすべての点を、あなたは関節の重みを見つけることができます。
最大値については、我々はそれに最大重量と二番目に大きい重みを見つける必要があります
やや面倒、第一関節重みは同時に、ノード内の任意の2つのノードに接続されているので、すべての点について、我々はそれだけでそれすべての重みの点および正方形を接続する必要があること、しかし我々は、彼らはそれが接続されているポイントのマイナス乗の和の値に右行くように、関節の重みと同じノードの2を選択することはできません。
コードは以下の通りであります:
#include<iostream>
#include<vector>
using namespace std;
int n;
long long ans,sum,a[200005];
vector<int>g[200005];
int main() {
cin>>n;
for(int i=1; i<n; i++) {
int u,v;
cin>>u>>v;
g[u].push_back(v);
g[v].push_back(u);
}
for(int i=1; i<=n; i++)cin>>a[i];
for(int u=1;u<=n;u++){
long long h1=0,h2=0,m1=0,m2=0;
for(int i=0;i<g[u].size();i++){
int v=g[u][i];
h1=h1+a[v],h2=h2+a[v]*a[v];
a[v]>m1?m2=m1,m1=a[v]:(a[v]>m2?m2=a[v]:0);
}
ans=max(ans,m1*m2);
sum=(sum+h1*h1-h2)%10007;
}
cout<<ans<<" "<<sum;
return 0;
}
時間の複雑さ:\(O(N)\)
100ポイント:スコアを期待