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("");
    }

}