bzoj3714: [PA2014]Kuglarz(最小生成樹)

Hanks_o發表於2018-03-21

題目傳送門

解法:
好神啊。
sum[i]表示前i個有多少個球。
那麼詢問一次i,j表示sum[i]-sum[j-1]的奇偶性。
那麼只要所有的sum都出現一次就絕對可以算出來了。
如果
問了3 5
問了6 8
那麼就不需要問3 8了。

所以。
最小生成樹呀!

程式碼實現:

#include<cstdio>
#include<iostream>
#include<algorithm>
using namespace std;
typedef long long ll;
struct node {int x,y;ll c;}a[5100000];int len;
bool cmp(node n1,node n2) {return n1.c<n2.c;}
int fa[2100];int findfa(int x) {if(fa[x]!=x)fa[x]=findfa(fa[x]);return fa[x];}
int main() {
    int n;scanf("%d",&n);len=0;
    for(int i=1;i<=n;i++)for(int j=i;j<=n;j++) {len++;a[len].x=i-1;a[len].y=j;scanf("%lld",&a[len].c);}
    sort(a+1,a+1+len,cmp);int t=0;for(int i=0;i<=n;i++)fa[i]=i;ll ans=0;
    for(int i=1;i<=len;i++) {
        int xx=findfa(a[i].x),yy=findfa(a[i].y);
        if(xx!=yy) {
            fa[xx]=yy;t++;ans+=a[i].c;if(t==n) {printf("%lld\n",ans);return 0;}
        }
    }printf("%lld\n",ans);
    return 0;
}

相關文章