CodeForces 123A 并查集

写了好长时间的代码，思路一看就有了，然后写了下，发现自己这方面的题目写的真是不好。。。擦。

题意：给一个字符串，然后对于这个字符串来说，他的素数位置 * i (1=<i <= len)要等于他本身，从新构造这个字符串，判断是否有一种情况满足这个条件。。

/* 这个代码自己写的真挫，有思路但是写不出来。。。。。*/
#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<queue>
using namespace std;
int father[10001];
int find(int x){
if(x == father[x])return x;
else return father[x] = find(father[x]);
}
int main()
{
char a[1009],b[1009];
cin >> (a+1);
int len = strlen(a+1);
int vis[1008];

for(int i=1;i<=len;i++)father[i] = i;
for(int i=2;i<=len;i++){
for(int j=i;j<=len;j+=i){
if(vis[j]){
int x = find(j);
father[i] = x;
}
else
{
father[j] =i;vis[j] = 1;
}

}
}
int numa[1008];
memset(numa,0,sizeof(numa));
for(int i=1;i<=len;i++)
numa[a[i]-'a']++;
int num[1008];
memset(num,0,sizeof(num));
int cn = 0;
int k[1008];
for(int i=1;i<=len;i++)
{
for(int j=1;j<=len;j++)
{
if(father[j] == i)num[i]++;
}
if(num[i])k[cn++] = i;
}
int flag = 1;
for(int i=0;i<cn;i++){
int t=k[i],Max = 10000000,pos =-1;
for(int j=0;j<=25;j++){
if(!numa[j])continue;
if(numa[j] - num[k[i]] < Max &&numa[j] >=num[k[i]]){
Max = numa[j] - num[k[i]];
t = k[i];
pos = j;
}
}
if(pos == -1){flag = 0;break;}
for(int h=1;h<=len;h++){
if(father[h] == t){
b[h] = pos+'a';
numa[pos] --;
}
}
num[t] = 0;
}

b[len+1] = '\0';
int cnt = 0;
if(flag == 0)puts("NO");
else{
puts("YES");
for(int i=1;i<=len;i++)
{
if(b[i] <'a'){
for(int j=0;j<=25;j++)
if(numa[j]>0)
{
printf("%c",j+'a');
numa[j]--;
}
}
else printf("%c",b[i]);
}
puts("");
}

}