Shichikuji and Power Grid 思路 最小生成树

题目传送门-->传送门

 

题意：给出n个点以及他们的坐标，每个点可以自己建造电力站，第i个点自造电力站的花费为c[i],也可以两点之间连电线，花费为电线的长度乘以两个点各自的k值相加。想让每个点都有电，问最小花费。

题解：两两点之间连线，权值为连电线的花费。在创建一个新的点0，连接上每个点，权值为每个点本身自造电力站的花费。建完图之后跑一遍最小生成树即可。

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cmath>
using namespace std;
#define ll long long
ll n, c, cnt;
struct node{
    ll x, y;
}point[3000], ans2[3000];
struct Edge{
    ll u, v, w;
    bool operator < (const Edge &t) const{
        return w<t.w;
    }
}e[5000000+20];
ll father[3000];
void init(){
    for(ll i=1; i<=n; i++)
        father[i]=i;
}
ll Find(ll x){
    if(x!=father[x]) father[x] = Find(father[x]);
    return father[x];
}
void Merge(ll a, ll b){
    ll p=Find(a);
    ll q=Find(b);
    if(p > q)
        father[p] = q;
    else father[q] = p;
}
void add(ll u, ll v, ll w){
    e[++cnt].u = u;
    e[cnt].v=v;
    e[cnt].w=w;
}
ll g[3000][3000];
ll k[3000], ans1[3000], cntans1, cntans2, sum;
void kruskal(ll n){
    ll k=0;
    ll ans=0;
    for(ll i=1; i<=cnt; i++){
        if(Find(e[i].u)!=Find(e[i].v)){
            Merge(e[i].u, e[i].v);
            if(e[i].u==0) ans1[++cntans1] = e[i].v;
            else ans2[++cntans2].x=e[i].u, ans2[cntans2].y=e[i].v;
            sum+=e[i].w;
            n--;
            if(n==0) return;
        }
    }
}
int main(){
    scanf("%I64d",&n);
    init();
    for(ll i=1; i<=n; i++)
        scanf("%I64d%I64d",&point[i].x,&point[i].y);
    for(ll i=1; i<=n; ++i) {
        scanf("%I64d",&c);
        add(0, i, c);
    }
    for(ll i=1; i<=n; i++)
        scanf("%I64d",&k[i]);
    for(ll i=1; i<=n; i++){
        for(ll j=1; j<=n; j++){
            if(i==j) add(i, j, 0);
            else {
                ll mi=(abs(point[i].x-point[j].x)+abs(point[i].y-point[j].y))*(k[i]+k[j]);
                add(i, j, mi);
            }
        }
    }
    sort(e+1, e+1+cnt);
    kruskal(n);
    printf("%I64d\n%I64d\n",sum,cntans1);
    for(ll i=1; i<=cntans1; i++)
        printf("%I64d%c",ans1[i],i==cntans1?'\n': ' ');
    printf("%I64d\n",cntans2);
    for(ll i=1; i<=cntans2; i++){
        printf("%I64d %I64d\n",ans2[i].x, ans2[i].y);
    }
    return 0;
}