洛谷 3141

【题目分析】
最小生成树的另类算法。

---

【代码】

#include <cstdio>
#include <cmath>
#include <cstring>
#include <iostream>
#include <algorithm>
#define ll long long
using namespace std;
int x[30001],y[30001];
int cntx,cnty,xx,yy,cnt=0,px=1,py=1;
long long ans=0;
int main()
{
    scanf("%d%d%d%d",&xx,&yy,&cntx,&cnty);
    for (int i=1;i<=cntx;++i) scanf("%d",&x[i]);x[++cntx]=xx;
    sort(x+1,x+cntx+1);
    for (int i=1;i<=cnty;++i) scanf("%d",&y[i]);y[++cnty]=yy;
    sort(y+1,y+cnty+1);
    for (int i=cntx;i>=1;--i) x[i]=x[i]-x[i-1];
    sort(x+1,x+cntx+1);
    for (int i=cnty;i>=1;--i) y[i]=y[i]-y[i-1];
    sort(y+1,y+cnty+1);
    int flagx=0,flagy=0,nowx=0,nowy=0;
    while (cnt<cntx*cnty-1)
    {
        if (x[px]<y[py])
        {
            if (!flagx||!flagy)
            {
                cnt+=cnty-1;
                ans+=(ll)(cnty-1)*x[px];
            }
            else
            {
                cnt+=cnty-nowx;
                ans+=(ll)(cnty-nowx)*x[px];
            }
            nowy++;
            flagy=1;
            px++;
        }
        else
        {
            if (!flagx||!flagy)
            {
                cnt+=cntx-1;
                ans+=(ll)(cntx-1)*y[py];
            }
            else
            {
                cnt+=cntx-nowy;
                ans+=(ll)(cntx-nowy)*y[py];
            }
            nowx++;
            flagx=1;
            py++;
        }
    }
    printf("%lld\n",ans);
}