UPC 6492（vector+sort）

题目链接：点击打开链接

题目大意：n个城市，k条公路，l条地铁，输出n个数字分别表示第i个城市与几个城市能既公路联通又地铁联通。

题目思路：刚开始用并查集，可是直接暴力查询n^2，不出意外的tle了，后来改用了邻接表，傻乎乎的没看到情空vis也用了n的复杂度....后来看了学姐的代码，对于每一组公路群以及地铁群（即互相之间都能联通）都给一个标号，然后根据这个标号进行排序，这个时候就可以直接比对，如果公路号和地铁号一样，那就美滋滋了。由于是先根据公路号排序，然后根据地铁号排序，所以如果公路号和地铁号一样那必然是会贴在一起的，实在太优秀了。

以下是代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
using namespace std;
#define MAXN 200005
struct node{
    int id,fir,sec;
}a[MAXN];
vector<int>v1[MAXN],v2[MAXN];
int cnt,vis[MAXN],ans[MAXN];
void dfs(int x){
    int len=v1[x].size();
    vis[x]=1,a[x].fir=cnt;
    for(int i=0;i<len;i++){
        if(vis[v1[x][i]])
        continue;
        dfs(v1[x][i]);
    }
    return;
}
void dfs2(int x){
    int len=v2[x].size();
    vis[x]=1,a[x].sec=cnt;
    for(int i=0;i<len;i++){
        if(vis[v2[x][i]])
        continue;
        dfs2(v2[x][i]);
    }
    return;
}
int cmp(node a,node b){
    if(a.fir==b.fir)
        return a.sec<b.sec;
    return a.fir<b.fir;
}
int main(){
    int n,k,l,x,y;
    while(~scanf("%d%d%d",&n,&k,&l)){
        for(int i=1;i<=n;i++){
            a[i].id=i;
            v1[i].clear();
            v2[i].clear();
        }
        for(int i=0;i<k;i++){
            scanf("%d%d",&x,&y);
            v1[x].push_back(y);
            v1[y].push_back(x);
        }
        for(int i=0;i<l;i++){
            scanf("%d%d",&x,&y);
            v2[x].push_back(y);
            v2[y].push_back(x);
        }
        cnt=0;
        memset(vis,0,sizeof(vis));
        for(int i=1;i<=n;i++){
            if(vis[i]){
                continue;
            }
            dfs(i);
            cnt++;
        }
        cnt=0;
        memset(vis,0,sizeof(vis));
        for(int i=1;i<=n;i++){
            if(vis[i]){
                continue;
            }
            dfs2(i);
            cnt++;
        }
        sort(a+1,a+n+1,cmp);
        for(int i=1;i<=n;i++){
            int j;
            for(j=i;j<=n;j++){
                if(a[i].fir!=a[j].fir||a[i].sec!=a[j].sec){
                    break;
                }
            }
            for(int k=i;k<j;k++){
                ans[a[k].id]=j-i;
            }
            i=j-1;
        }
        for(int i=1;i<=n;i++){
            printf("%d%c",ans[i],i==n?'\n':' ');
        }
    }
    return 0;
}