程序自动分析【离散化+并查集】

Alan_Lowe

已于 2022-03-02 22:30:44 修改

阅读量224

点赞数 1

分类专栏： # 并查集、最小生成树 # 离散化、分治文章标签：开发语言算法数据结构

于 2022-03-02 22:30:16 首次发布

本文链接：https://blog.csdn.net/qq_45985728/article/details/123242165

版权

并查集、最小生成树同时被 2 个专栏收录

16 篇文章 0 订阅

订阅专栏

离散化、分治

8 篇文章 0 订阅

订阅专栏

该博客主要讲解如何运用离散化和并查集算法解决NOI2015中的一道题目。首先，通过离散化对输入数据进行排序和去重，然后利用并查集实现集合合并和查询操作。在求解过程中，通过并查集优化路径压缩，确保查询效率。最后，通过实例演示了算法的具体应用和代码实现。

摘要由CSDN通过智能技术生成

程序自动分析【离散化+并查集】

NOI2015、BZOJ4195、ACwing237

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

#define int long long
#define pii pair<int,int>
const int N = 200000;

int father[N + 5],a[N + 5],b[N + 5];    //分别用来存储离散化后每个结点的代标结点、初始输入点、去重后的点
int n,k_a,k_b;              //n如题，
vector<pii> v0,v1;          //相等的组、不相等的组

int find_f(int x){          //找到祖先结点/集合中的代表结点
    if(father[x] == x)
        return x;
    return father[x] = find_f(father[x]);   //缩短路径
}
void discrete(){            //离散化
    sort(a + 1,a + k_a + 1);
    b[++k_b] = a[1];
    for(int i = 2;i <= k_a;++i){
        if(a[i] != a[i - 1])
            b[++k_b] = a[i];
    }
}
int find_w(int x){          //找到离散化后代替值x的数。
    return lower_bound(b + 1,b + k_b + 1,x) - b;
}
void solve(){
    k_a = k_b = 0;v0.clear();v1.clear();    //每次都要初始化一下
    for(int i = 1;i <= N;++i)               //初始化father数组
        father[i] = i;
    cin>>n;
    int x,y,z;
    for(int i = 1;i <= n;++i){
        cin>>x>>y>>z;
        a[++k_a] = x;a[++k_a] = y;
        if(z)
            v1.push_back(pii(x,y));     //z等于1的时候要合并的情况
        else
            v0.push_back(pii(x,y));     //z等于0的时候要查询的情况
    }
    discrete();                         //离散化
                                        //合并
    for(vector<pii>::iterator it = v1.begin();it != v1.end();++it){
        father[find_f(find_w(it->first))] = find_f(find_w(it->second));
    }
                                        //查询
    for(vector<pii>::iterator it = v0.begin();it != v0.end();++it){
        if(find_f(find_w(it->first)) == find_f(find_w(it->second))) {
            cout << "NO\n";
            return;
        }
    }
    cout<<"YES\n";
}

signed main(){
    ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
    int tt;
    cin>>tt;
    while(tt--)
        solve();
    return 0;
}