1. 程式人生 > >資料結構與演算法C++之並查集

資料結構與演算法C++之並查集

並查集 Union Find
可以解決連線問題和路徑問題
在這裡插入圖片描述
下圖中使用id表示陣列,第一行表示其索引,第二行表示元素值,為0的元素表示是互相連線在一起的,同樣為1的元素表示互相連線在一起的
在這裡插入圖片描述
在這裡插入圖片描述
下面是測試一個並查集 Union 操作的計算複雜度

#include <iostream>
#include <cassert>
#include "UnionFindTestHelper.h"

using namespace std;

int main()
{
    int n = 100000;
    UnionFindTestHelper::testUF1(n);

    return
0; }

"UnionFindTestHelper.h"為

//UnionFindTestHelper.h
#include <iostream>
#include <ctime>
#include <cstdlib> //rand()函式
#include "UnionFind1.h"

using namespace std;

namespace UnionFindTestHelper{

    void testUF1(int n){
        srand(time(NULL));
        UF1::UnionFind uf = UF1::
UnionFind(n); time_t startTime = clock(); for(int i = 0; i < n; i++){ int a = rand()%n; int b = rand()%n; uf.unionElements(a, b); } for(int i = 0; i < n; i++){ int a = rand()%n; int b = rand()%n; uf.
isConnected(a, b); } time_t endTime = clock(); cout<<"UF1, "<<2*n<<" ops, "<<double(endTime-startTime)/CLOCKS_PER_SEC<<" s"<<endl; } }

"UnionFind1.h"為

#include <iostream>
#include <cassert>

using namespace std;

namespace UF1 {

    class UnionFind {

    private:
        int *id;
        int count;

    public:
        UnionFind(int n){
            count = n;
            id = new int[n];
            for(int i = 0; i < n; i++){
                id[i] = i;
            }
        }

        ~UnionFind(){
            delete[] id;
        }

        int find( int p){
            assert( p >= 0 && p < count );
            return id[p];
        }

        bool isConnected(int p, int q){
            return find(p) == find(q);
        }

        void unionElements(int p, int q){
            int pID = find(p);
            int qID = find(q);

            if ( pID == qID )
                return;

            for(int i = 0; i < count; i++){
                if( id[i] == pID )
                    id[i] = qID;
            }
        }
    };
}

輸出為
在這裡插入圖片描述
可以看出執行20萬個操作用時9秒多,時間還是很長的
由於for迴圈遍歷一遍n個元素,複雜度為O(n)
unionElements,複雜度為O(n),
所以最終複雜度為O(n2),當然就慢啦