鏈接：https://www.nowcoder.com/acm/contest/133/A
來源：牛客網
?

時間限制：C/C++ 1秒，其他語言2秒
空間限制：C/C++ 262144K，其他語言524288K
64bit IO Format: %lld

題目描述

現在有一棵被Samsara-Karma染了k種顏色的樹，每種顏色有著不同的價值

Applese覺得Samsara-Karma染的太難看了，于是打算把整棵樹重新染成同一種顏色

但是，由于一些奧妙重重的原因，每一次染色Applese可以選擇兩個有邊相連的點，將其中一個染成另一個的顏色。而進行一次這樣的操作需要付出兩種顏色價值和的代價

現在，Applese的錢要用來買書(game)，所以他想要最小化代價

?

輸入描述:

輸入包括若干行 第一行包括一個數n，表示這棵樹有n個節點 第二行包括n個數，第i個數表示第i個節點的顏色coli **注意：一個顏色的標號即價值

接下來的n - 1行，每行包括兩個數u, v，表示u節點與v節點之間有一條無向邊

n ≤ 100000, 1 ≤ coli ≤ 1e9，數據保證是一棵樹

輸出描述:

輸出包括一行 第一行包括一個數，表示最小代價

?

示例1

輸入

4 2 3 4 3 1 2 2 3 3 4

輸出

12

?

最開始暴力過去的，數據比較弱。強行枚舉所有顏色的花費。時間復雜度O(n^2)

#include <bits/stdc++.h> using namespace std; const int MAXN=1e5+10; int a[MAXN]; map<int, int >ma; set<int>se; int x[MAXN],y[MAXN]; int main() {int n;scanf("%d",&n);for (int i = 1; i <=n ; ++i) {scanf("%d",&a[i]);se.insert(a[i]);}for (int i = 1; i <n ; ++i) {scanf("%d%d",&x[i],&y[i]);}long long ans=1e14,sum=0;set<int>::iterator it;for (it=se.begin(); it !=se.end() ; ++it) {sum=0;for (int j = 1; j <=n ; ++j) {if((*it)!=a[j]) sum+=((*it)+a[j]);}ans=min(sum,ans);}printf("%lld\n",ans);return 0; }

思維：時間復雜度O(n)

#include<bits/stdc++.h> using namespace std; #define LL long longconst int MAXN = 1e5+10; const int INF = 0x3f3f3f3f; int n,u,v; int a[MAXN]; map<int,LL>mp,cost; int main(){while(scanf("%d",&n)!=EOF){LL sum = 0;for(int i=1;i<=n;i++){scanf("%d",&a[i]);mp[a[i]] += a[i];sum += a[i];}for(int i=1;i<n;i++){scanf("%d%d",&u,&v);}LL ans = 1LL<<60;map<int,LL>::iterator it;for(it=mp.begin();it!=mp.end();it++){ans = min(ans,sum - it->second + (n - it->second / it->first) * it->first);}cout<<ans<<endl;}return 0; }

?

總結

以上是生活随笔為你收集整理的Wannafly挑战赛20 A 染色（暴力/思维）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。