Codeforces 1245 D. Shichikuji and Power Grid

传送门

经典的最小生成树模型

建一个点 $0$ ，向所有其他点 $x$ 连一条边权为 $c[x]$ 的边，其他任意两点之间连边，边权为 $(k_i+k_j)(\left | x_i-x_j\right |+\left | y_i-y_j\right |)$

然后用 $prim$ 求个最小生成树即可，然后考虑一下输出方案

多维护一个 $frm[x]$ 表示当前的 $dis[x]$ 是从哪个点贡献来的就行了

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<vector>
using namespace std;
typedef long long ll;
inline int read()
{
    int x=,f=; char ch=getchar();
    while(ch<''||ch>'') { if(ch=='-') f=-; ch=getchar(); }
    while(ch>=''&&ch<='') { x=(x<<)+(x<<)+(ch^); ch=getchar(); }
    return x*f;
}
const int N=;
const ll INF=1e18;
int n,x[N],y[N],c[N],k[N];
inline ll cst(int i,int j) { return 1ll*(k[i]+k[j])*(abs(x[i]-x[j])+abs(y[i]-y[j])); }
ll dis[N],ans;
bool vis[N];
vector <int> bas;
vector <pair<int,int>> edg;
int frm[N];
void Prim()
{
    for(int i=;i<=n;i++) dis[i]=c[i];
    for(int I=;I<=n;I++)
    {
        ll mx=INF; int pos=-;
        for(int i=;i<=n;i++)
            if(!vis[i]&&dis[i]<mx)
                mx=dis[i],pos=i;
        vis[pos]=; ans+=mx;
        if(!frm[pos]) bas.push_back(pos);
        else edg.push_back(make_pair(frm[pos],pos));
        for(int i=;i<=n;i++)
            if(!vis[i]&&dis[i]>cst(pos,i))
                dis[i]=cst(pos,i),frm[i]=pos;
    }
}
int main()
{
    n=read();
    for(int i=;i<=n;i++)
        x[i]=read(),y[i]=read();
    for(int i=;i<=n;i++) c[i]=read();
    for(int i=;i<=n;i++) k[i]=read();
    Prim(); printf("%lld\n",ans);
    printf("%d\n",int(bas.size()));
    for(auto x: bas) printf("%d ",x); puts("");
    printf("%d\n",int(edg.size()));
    for(auto t: edg) printf("%d %d\n",t.first,t.second);
    return ;
}