【蓝桥杯】2017年第八届真题-合根植物

c++_learner

于 2021-10-09 09:53:13 发布

阅读量161

点赞数

文章标签：数据结构

本文链接：https://blog.csdn.net/qq_52103578/article/details/120665011

版权

本文介绍了一个使用C++编写的程序，通过预处理和并行连接算法，解决植物关系问题。它利用unordered_set实现桶操作，计算并返回经过特定关系调整后的植物数量。核心内容包括初始化、连接节点和find操作的运用。

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

#include<iostream>
#include<vector>
#include<utility>
#include<unordered_set>
using namespace std;

class Plants{
    vector<int> pre;
public:
    Plants(){
        pre.resize(1000001);
    }
    int getNum(){
        int m,n,k;
        cin>>m>>n;
        cin>>k;
        vector<pair<int,int>> relation(k);
        for(int i=0;i<k;i++){
            cin >> relation[i].first >> relation[i].second;
        }
        int originalNum = m*n;
        init (originalNum);
        for(int i=0;i<k;i++){
            join(relation[i].first,relation[i].second);
        }
        unordered_set<int> bucket;
        for(int i=1;i<=originalNum;i++){
            int a = find(i);
            if(bucket.find(a)==bucket.end()) bucket.insert(a);
        }
        return bucket.size();
    }
    void init(int n){
        for(int i=1;i<=n;i++){
            pre[i]=i;
        }
    }
    void join(int i,int j){
        int a=find(i),b=find(j);
        if(a==b) return;
        pre[a] = b;
    }
    int find(int i){
        if(i==pre[i]) return i;
        return pre[i]=find(pre[i]);
    }
};
int main()
{
    Plants solution;
    cout << solution.getNum() << endl;
    return 0;
}