資料結構與演算法C++之並查集
阿新 • • 發佈:2018-12-22
並查集 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),當然就慢啦