D. Shichikuji and Power Grid

[Shichikuji and Power Grid]添加链接描述
codeforces 597 div2 的D 题，
题意：在一个二维平面上面，有n个城市，现在每个城市都没有电。
你可以选择一些城市建发电站，代价是c[i]；你也可以给每个城市拉电线，给城市(i,j)之间拉电线的代价是(abs(x[i]-x[j])+abs(y[i]-y[j]))*(k[i]+k[j])。
现在问你最少花费多少代价，能够使得全部城市都有电，输出方案。

思路：写的时候无限自闭，最后A了（感谢龙佬），其实思路很简单，就是把这个题变成最小生成树问题就行了，相当于把供电站自身的花费变成了超级远点到供电站的距离，然后套一下最小生成树模板，就行了

#include<bits/stdc++.h>
using namespace std;
const int maxn=2e6+10;
#define ll long long 
ll f[maxn],r[maxn];
typedef pair<ll,ll> pil;
struct node {
	ll x;int y;	
}p[maxn];
struct Node {
	ll u, v, w;
}P[maxn];
bool cmp(Node a,Node b)
{
	return a.w<b.w;
}
bool cmp2(pil a,pil b)
{
	return a.first <b.first;
}
ll find(ll x)
{
	return f[x]!=x?f[x]=find(f[x]):x;
}
void join(int x,int y)
{
	x=find(x);
	y=find(y);
	if(x!=y)
	f[x]=y;
}
 
vector<pil> s,ss;
int main(){
	ll n,pos,co;
	cin>>n;
	for(int i=0;i<=n;i++) f[i]=i;
    for(int i=1;i<=n;i++) cin>>p[i].x>>p[i].y;	
	for(int i=1;i<=n;i++)
	{ 
	   cin>>co;P[i].u=i;P[i].v=0;P[i].w=co;
	}
	for(int i=1;i<=n;i++) cin>>r[i];
	pos=n+1;
	for(int i=1;i<=n;i++)
	for(int j=i+1;j<=n;j++)
	{
	  ll mid=(abs(p[i].x-p[j].x)+abs(p[i].y-p[j].y))*(r[i]+r[j]);
	  P[pos].u=j,P[pos].v=i,P[pos++].w=mid;	
	}
	ll sum=0,a=0,b=0;
	sort(P+1,P+pos+1,cmp);
	for(int i=1;i<=pos;i++)
	{
		if(find(P[i].u)!=find(P[i].v))
		{
			join(P[i].u,P[i].v);
			sum+=P[i].w;
			if(P[i].u!=0&&P[i].v!=0)
			a++,s.push_back(make_pair(P[i].u,P[i].v));
			else b++,ss.push_back(make_pair(P[i].u,P[i].v));
			
		}
	}
	cout<<sum<<endl;
	cout<<b<<endl;
	sort(ss.begin(),ss.end(),cmp2);
	for(int i=0;i<ss.size();i++)
	{  ll a1 = ss[i].first, a2 = ss[i].second;
		if(a1)
		cout<<a1<<" ";
		else 
		cout<<a2<<" "; 
	}
	cout<<endl;
	cout<<a<<endl;
	for(int i=0;i<s.size();i++)
	{
	  ll a1 = s[i].first, a2 = s[i].second;
        if (a1 > a2)
        swap(a1, a2);
        cout << a1 << " " << a2 << endl;
	}
}