（Codeforces Round 597 (Div. 2)）D. Shichikuji and Power Grid

Problem - 1245D - Codeforces

思路：最小生成树，因为题目给出点的范围为2000，而所有的边权我们都可以得到，所以先进行暴力建边，然后设置一个不存在的点，将该点和所有点建立一条边权为ci的边，之后跑最小生成树即可。

#include<bits/stdc++.h>

using namespace std;
const int N=2e3+5;
typedef long long ll;
typedef pair<ll,ll> pll;
int mod=1e9+7;
const int maxv=4e6+5;
struct node
{
	ll a,b,w;
}e[maxv];
int p[N];
int find(int x)
{   
    if(p[x]!=x){
        return p[x]=find(p[x]);
    }
    return x;
}
void solve()
{	
	int n;

	cin>>n;
	vector<pll> v(n+1);
	vector<int> k(n+1);
	vector<int> c(n+1);
	for(int i=1;i<=n;i++){
		int a,b;
		cin>>a>>b;
		v[i]={a,b};
		p[i]=i;
	}
	for(int i=1;i<=n;i++) cin>>c[i];
	for(int i=1;i<=n;i++) cin>>k[i];
	int cnt=0;
	for(int i=1;i<=n;i++){//建立一个不存在的点连向所有边
		e[cnt++]={0,i,c[i]};
	}
	p[0]=0;
	for(int i=1;i<=n;i++){//建边
		for(int j=i+1;j<=n;j++){
			ll x1=v[i].first,y1=v[i].second;
			ll x2=v[j].first,y2=v[j].second;
			ll val=(ll)(k[i]+k[j])*(abs(x1-x2)+abs(y1-y2));//计算边权
			e[cnt++]={i,j,val};
		}
	}
	sort(e,e+cnt,[](node x,node y){//最小生成树按边权升序排序
		return x.w<y.w;
	});
	ll res=0;
	vector<pll> ans;
	vector<int> z;
	for(int i=0;i<cnt;i++){//最小生成树
		int a=e[i].a,b=e[i].b;
		ll w=e[i].w;
		int pa=find(a),pb=find(b);
		if(pa!=pb) {
			p[pa]=pb;
			res+=w;
			if(a==0) z.push_back(b);//判断需要被点亮的点
			else{
				ans.push_back({a,b});//正常连接的点
			}
		}
	}
	cout<<res<<endl;
	cout<<z.size()<<endl;
	for(auto x: z){
		cout<<x<<" ";
	}
	cout<<endl;
	cout<<ans.size()<<endl;
	for(auto [x,y]: ans){
		cout<<x<< " "<<y<<endl;
	}
}

int main()
{
	ios::sync_with_stdio(0);
	cin.tie(0);
	cout.tie(0);
	int t;
	t=1;
	//cin>>t;
	while(t--){
		solve();
	}
	system("pause");
	return 0;
}