AtCoder Regular Contest 097

因为太菜了所以只有前两题>_<

C.
题意：给定一个$|S|<=5000$的字符串$S$，求它的所有子串（不重复）中第k小的$(k<=5)$
题解：很有意思，我们只需要找到$S$的所有长度小于等于5的子串然后排序取第k个就好了，时间复杂度也不大OwO

#include<bits/stdc++.h>
#define LiangJiaJun main
using namespace std;
char s[5004];
int k,l;
set<string>tr;
int LiangJiaJun(){
    scanf("%s",s+1);
    l=strlen(s+1);
    int cnt=0;
    scanf("%d",&k);
    for(int i=1;i<=5;i++){
        for(int j=1;j+i-1<=l;j++){
            string m="";
            for(int t=j;t<=j+i-1;t++){
                m=m+s[t];
            }
            tr.insert(m);
        }
    }
    string s;
    for(set<string>::iterator it=tr.begin();it!=tr.end()&&k>0;it++,k--){
        s=(*it);
    }
    cout<<s<<endl;
    return 0;
}

D.
题意：给一个n的排列$p1,p2,p3,...,pn$，和m条关系$(0<n,m<=10^5)$,每一条关系有两个数字$(x,y)$ 表示该操作可以交换位置x和位置y的数字。求问你通过一系列操作之后(操作次数不限)，符合$pi=i$的情况的$i(1<=i<=n)$有多少个?

题解：我们可以知道，对于一个联通块内所有点他们都是可以任意换位置的，那么我们对于每一个联通块内使得联通块内的$pi=i$组数最多就OK啦。用并查集实现联通块的查找。

#include<bits/stdc++.h>
#define LiangJiaJun main
using namespace std;
int n,m;
int f[100004],a[100004];
int Find(int x){return f[x]==x?x:f[x]=Find(f[x]);}
vector<int>G[100004];
set<int>tr[100004];
int LiangJiaJun(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)f[i]=i;
    for(int i=1;i<=n;i++)scanf("%d",&a[i]);
    for(int i=1;i<=m;i++){
        int u,v,p,q;
        scanf("%d%d",&u,&v);
        p=Find(u);q=Find(v);
        if(p!=q)f[p]=q;
    }
    int ans=0;
    for(int i=1;i<=n;i++){
        tr[Find(i)].insert(i);
    }
    for(int i=1;i<=n;i++){
        int tg=Find(i);
        if(tr[tg].find(a[i])!=tr[tg].end())++ans;
    }
    printf("%d\n",ans);
    return 0;
}