有 n n n 个杯子,每个杯子最多一个球,你可以用 c i , j c_{i,j} ci,j 元得知 [ i , j ] [i,j] [i,j] 中杯子球的个数,问最少花费多少钱可以得知所有杯子的情况。 n ≤ 2 × 1 0 3 n\leq 2\times 10^3 n≤2×103 。
首先定义 d i , j = 1 d_{i,j}=1 di,j=1 为知道 [ i , j + 1 ] [i,j+1] [i,j+1] 的奇偶情况,那么有 d i , j , d j , k , d i , k d_{i,j},d_{j,k},d_{i,k} di,j,dj,k,di,k 其中2个值为1,即可得到第三个值为1,关系具有传递性。同时,根据定义 d i , i + 1 = c i , i d_{i,i+1}=c_{i,i} di,i+1=ci,i ,即若所有 d i , i + 1 = 1 d_{i,i+1}=1 di,i+1=1 就得到答案。那么建出图来,对于每一组 c i , j c_{i,j} ci,j ,给定将 i i i 和 j + 1 j+1 j+1 建一条长度 c i , j c_{i,j} ci,j 的边,在图上跑最小生成树即可。
#include<bits/stdc++.h>
using namespace std;
#define int long long
struct edge{
int u,v,w;
}e[4000005];
int fa[5005],n,ans,eu,ev,cnt,ecnt;
bool cmp(edge a,edge b){
return a.w<b.w;
}
int find(int x){
return (fa[x]==x)?x:fa[x]=find(fa[x]);
}
void kruskal(){
sort(e+1,e+ecnt+1,cmp);
for(int i=1;i<=ecnt;i++){
eu=find(e[i].u), ev=find(e[i].v);
if(eu==ev){
continue;
}
ans+=e[i].w;
fa[ev]=eu;
if(++cnt==n-1){
break;
}
}
}
signed main(){
scanf("%lld",&n);
for(int i=1;i<=n+1;i++){
fa[i]=i;
}
for(int i=1;i<=n;++i){
for(int j=i;j<=n;++j){
int sij;
scanf("%lld",&sij);
e[++ecnt].u=i;
e[ecnt].v=j+1;
e[ecnt].w=sij;
}
}
++n;
kruskal();
printf("%lld",ans);
return 0;
}