pat中关于并查集的题目&&以及类似dfs解法

点击打开链接：https://www.patest.cn/contests/pat-a-practise/1013

并查集：主要是find和union函数：

void unionn(int a,int b){
    int x=find(a);
    int y=find(b);
    if(x!=y){
        pre[x]=y;
    }
}

int find(int k){
    if(pre[k]!=k){
        return pre[k]=find(pre[k]);
    }
    else return k;
}

这道题其实就是在删除某个节点相关的所有路径之后用并查集找出有多少个独立的分支，也就是pre[i]==i的数量

//
//  main.cpp
//  1013
//
//  Created by DouglasLee on 18/3/17.
//  Copyright © 2018年 DouglasLee. All rights reserved.
//

#include <iostream>
using namespace std;
int road[1000000][2];
int pre[1005];
int find(int k){
    if(pre[k]!=k){
        return pre[k]=find(pre[k]);
    }
    else return k;
}
void unionn(int a,int b){
    int x=find(a);
    int y=find(b);
    if(x!=y){
        pre[x]=y;
    }
}
int n,m,k,a;
int main(int argc, const char * argv[]) {
    cin>>n>>m>>k;
    for(int i=0;i<m;i++){
        int p,q;
        scanf("%d%d",&p,&q);
        road[i][0]=p;
        road[i][1]=q;
    }
    for(int i=0;i<k;i++){
        scanf("%d",&a);
        for(int j=0;j<1005;j++)
            pre[j]=j;
        for(int z=0;z<m;z++){
            if(road[z][0]!=a&&road[z][1]!=a)
                unionn(road[z][1],road[z][0]);
        }
        int cnt=0;
        for(int y=1;y<=n;y++){
            if(pre[y]==y&&y!=a)
                ++cnt;
        }
        cout<<--cnt<<endl;
    }
    return 0;
}

pat1034:https://www.patest.cn/contests/pat-a-practise/1034

这道题就是要你找出一个独立的团，并且这个团中的联系权重要大于某个阈值。这个可以用一下dfs，搜出一个节点能找出的整个集合，并确定出这个集合中最大权重的点。

这里需要注意的是map的用法，find函数如果找不到，就会返回map<,>.end()。

还有map插入的时候就直接map[s]=1的类似样子

另外遍历的话可以选择如下操作：

map<string,int>::iterator it=visit.begin();
    for(;it!=visit.end();it++){
//code
} 

 具体题解如下：

//
//  main.cpp
//  1034new
//
//  Created by DouglasLee on 18/3/17.
//  Copyright © 2018年 DouglasLee. All rights reserved.
//

#include <iostream>
#include<map>
#include<vector>
#include<algorithm>
#include<string>
using namespace std;
map<string,int> weight;
map<string,int> visit;
map<string,vector<string>> adj;
int total;
int cnt;
string head;
struct person{
    string name;
    int num;
};
vector<person> bangHead;
bool cmp(person p1,person p2){
    return p1.name<p2.name;
}
void dfs(string s){
    visit[s]=1;
    total+=weight[s];
    cnt++;
    if(weight[head]<weight[s])
        head=s;
    int size=adj[s].size();
    for(int i=0;i<size;i++){
        if(visit[adj[s][i]]==0)
            dfs(adj[s][i]);
    }
}
int main(int argc, const char * argv[]) {
    int n,k;
    cin>>n>>k;
    for(int i=0;i<n;i++){
        string s1,s2;
        int weigh;
        cin>>s1>>s2>>weigh;
        adj[s1].push_back(s2);
        adj[s2].push_back(s1);
        if(weight.find(s1)==weight.end())
            weight[s1]=0;
        if(weight.find(s2)==weight.end())
            weight[s2]=0;
        if(visit.find(s1)==visit.end())
            visit[s1]=0;
        if(visit.find(s2)==visit.end())
            visit[s2]=0;
        weight[s1]+=weigh;
        weight[s2]+=weigh;
    }
    map<string,int>::iterator it=visit.begin();
    for(;it!=visit.end();it++){
        string name=it->first;
        if(it->second==0){
            total=0;
            cnt=0;
            head=name;
            dfs(name);
            if(total>(2*k)&&cnt>2){
                person tmp;
                tmp.num=cnt;
                tmp.name=head;
                bangHead.push_back(tmp);
            }
        }
    }
    cout<<bangHead.size()<<endl;
    sort(bangHead.begin(),bangHead.end(),cmp);
    for(int i=0;i<bangHead.size();i++){
        cout<<bangHead[i].name<<' '<<bangHead[i].num<<endl;
    }
    return 0;
}