dsu题集

 

【带权并查集】CodeForces - 982D Shark

题目大意：

题意有点难理解，就是给你一个长为n的数组,你可以任意选定一个数字k，使得大于等于k的数变成0，其他数变成1,然后要满足每个1的块其中1的数量相同，并且你要保证为1的块尽可能多，问你满足条件的最小的k为多少。

解题分析：

因为该序列所有元素都不相同，所以最小的k值一定是其中某个元素的值+1，然后就是将序列元素的值从小到大排序后，枚举k的值，用带权并查集判断是否符合题意。

#include <bits/stdc++.h>
using namespace std;

const int N = 1e5+5;
struct Node{ int a,b; bool operator < ( const Node &tmp)const{ return b<tmp.b; } }node[N];

int n,cnt,mxcnt,ans;
int arr[N],fa[N],num[N],sum[N];

int find(int x){ return x==fa[x]?x:fa[x]=find(fa[x]); }

inline void Union(int x,int y){
    x=find(x),y=find(y);
    sum[num[x]]--,sum[num[y]]--;      //表示对应集合数量的统计--
    num[x]+=num[y];sum[num[x]]++;     //y集合合并到x集合上
    fa[y]=x;cnt--;            //cnt表示当前集合的数量
}
int main(){
    cin>>n;
    for(int i=1;i<=n;i++){
        scanf("%d",&arr[i]);
        node[i].b=arr[i];node[i].a=i;
        fa[i]=i;
    }
    sort(node+1,node+n+1);        
    for(int i=1;i<=n;i++){     //按序列中元素的大小从小到大考虑,枚举第i个元素，表示将k设为node[i].b+1
        int loc=node[i].a,val=node[i].b;
        num[loc]++;sum[num[loc]]++;   //初始化当前元素所在集合      
        cnt++;
        if(loc!=n&&val>arr[loc+1])Union(loc,loc+1); //因为当前k是设为node[i].b+1,所以该元素与原序列两边的元素比较，如果大于该元素，则说明他们在同一集合
        if(loc!=1&&val>arr[loc-1])Union(loc,loc-1);
        if(cnt==sum[num[find(loc)]]){       //判断是否所有集合的元素数量相同(判断是否满足题目条件)
            if(cnt>mxcnt)mxcnt=cnt,ans=val+1;
        }
    }
    printf("%d\n",ans);
}

View Code

 

转载于:https://www.cnblogs.com/00isok/p/11178742.html