BZOJ3714: [PA2014]Kuglarz

BZOJ3714: [PA2014]Kuglarz

https://lydsy.com/JudgeOnline/problem.php?id=3714

分析:

• 询问\(l\sim r\)转化成\(l-1\)和\(r\)。
• 新加一个\(0\)，把询问看成连边就是要求一棵最小生成树。
• prim即可

代码:

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cstdlib>
#include <set>
#include <vector>
#include <cmath>
using namespace std;
#define N 2050
typedef long long ll;
int n,a[N][N];
char buf[100000],*p1,*p2;
int dis[N],vis[N];
ll ans;
#define nc() (p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++)
int rd() {
    int x=0; char s=nc();
    while(s<'0') s=nc();
    while(s>='0') x=(((x<<2)+x)<<1)+s-'0',s=nc();
    return x;
}
int main() {
    n=rd();
    int i,j;
    for(i=1;i<=n;i++) {
        for(j=i;j<=n;j++) {
            a[i-1][j]=rd(); a[j][i-1]=a[i-1][j];
        }
        dis[i]=0x3f3f3f3f;
    }
    int x=0,nxt=0;
    vis[0]=1;
    for(i=1;i<=n;i++) {
        nxt=0;
        for(j=1;j<=n;j++) {
            if(!vis[j]) {
                dis[j]=min(dis[j],a[x][j]);
                if((!nxt)||dis[j]<dis[nxt]) nxt=j;
            }
        }
        x=nxt; ans+=dis[x]; vis[x]=1;
    }
    printf("%lld\n",ans);
}