并查集学习笔记

并查集

处理不相交集合的合并问题，：求连通子图，最小生成树（Kruskal），最近公共祖先（LCA）， LC2421好路径的数目 、LC2382删除操作后的最大子段和

初始化：一维数组存父节点（代表元），初始为自己。fa[N]

查询：递归不断往上查找祖先

//查找i的祖先
int find(int i){
    if(i != fa[i])
        fa[i] = find(fa[i]);
    return fa[i];
}

合并：合并两个元素，找到这两个元素的祖先，改变祖先合并成一连

//合并
void unionn(int i, int j){
    int i_fa = find(i);
    int j_fa = find(j);
    fa[i_fa] = j_fa;
}

 LC2421好路径的数目  

题目：给N个节点对应的值，给N条边。求好路径的数目（头尾两端是相同值，而且是路径中最大值）。

思路：并查集思想：初始化fa[i]数组存i的代表元，以及创建存放区间里相同值的个数s[i]=1表示i节点作为代表源的相同值的节点数（路径数等于两个集合相同值的数的乘积），将节点值从小到大排序，依次加入节点x，遍历所有相邻边，相邻节点是相同集合或值更大就跳过，值相同计算中间结果（包括res以及相同值的节点数），且更新到x里（因为x一定是集合里最大的），此时小于等于的节点都进行合并到x里。

int numberOfGoodPaths(vector<int>& vals, vector<vector<int>>& edges) {
    int n = vals.size();
    int res = n;
    //建图
    vector<vector<int>> graph(n);
    for(auto e:edges){
        graph[e[0]].push_back(e[1]);
        graph[e[1]].push_back(e[0]);
    }
    //size[i]数组表示连通块内he vals[i]的相同值的数量
    vector<int> s(n,1);
    vector<pair<int,int>> paixu;
    for(int i = 0; i < n; ++i){
        paixu.push_back({vals[i],i});
        fa.push_back(i);//并查集初始化
    }
    sort(paixu.begin(),paixu.end());//默认对first升序
    //按节点值从小到大排序依次取出
    for(auto node : paixu){
        int fx = find(node.second);//当前节点的代表元
        for(auto e:graph[node.second]){
            int fy = find(e);//边节点的代表元
            if(fy == fx || vals[fy] > vals[node.second]) continue;//相同代表元或值更大
            if(vals[fy] == vals[node.second]){
                res += s[fy]*s[fx];
                s[fx] += s[fy];
            }
            fa[fy] = fx;//fx一直都是最大的
        }
    }
    return res;
}

 LC2382删除操作后的最大子段和

题目：给正数数组nums，和删除步骤数组removeQueries。求每次删除后的最大子数组（连续区间）和。

思路：逆向思维变成添加，初始化fa[i]数组存i的代表元，以及创建存放区间结果的数组tmp{0}。每次添加一个元素x，把合并到x+1处，计算tmp[fa[x+1]] += nums[x] + tmp[x]。(tmp[x]是左边相邻的子集和，tmp[fa[x+1]]有右边的子集和)，计算后的结果即为当前操作所求。

（合并方法很巧妙，x直接合并到x+1。这样的话，如果之前x-1已经添加了，tmp[x]就已经加上nums[x-1]了；如果x+1之前已经添加过了，也已经被添加到tmp[fa[x+1]]里了）。

vector<long long> maximumSegmentSum(vector<int>& nums, vector<int>& removeQueries) {
    //fa[i]存i的代表元
    vector<long long> tmp(nums.size()+1);
    for(int i = 0; i < nums.size()+1; ++i){
        fa.push_back(i);
    }
    vector<long long> res(nums.size(),0);//要的是删除后的结果
    for(int i = nums.size()-1; i >= 1; --i){//最后一次删除必为0，第一次前删除也不用求
        int x = removeQueries[i];//当前添加元素的下标
        int to = find(x+1);
        fa[x] = to;//合并到x+1上
        tmp[to] += tmp[x] + nums[x];//左中有求和，都存在最右边的节点上。
        res[i-1] = max(res[i],tmp[to]);//与前一次对比，当前有可能是新的小子集。
    }
    return res;
}