Wannfly挑战赛20 A 染色

链接：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>
#define  ll long long
using namespace std;
set<ll> l;
set<ll>::iterator it;
ll a[100005];
int main()
{
    int n,v,c;
    cin>>n;
    for(int i=0;i<n;i++)
    {
        scanf("%lld",&a[i]);
        l.insert(a[i]);
    }
    for(int i=0;i<n-1;i++)
        scanf("%lld %lld",&c,&v);
    long long ans=0;
    long long ams;
 
    for( it=l.begin();it!=l.end();it++)
    {
        ans=0;
        for(int j=0;j<n;j++)
        {
            if(*it!=a[j])
                ans+=a[j]+*it;
            else continue;
        }
        ams=min(ans,ams);
    }
    printf("%lld\n",ams);
    return 0;
}

O(n)

#include<bits/stdc++.h>
using namespace std;
#define LL long long

const 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;
}