CF 124C C. Prime Permutation 暴力

题意:长度为n的字符串s,问是否能将s重新排列
满足对任意一个素数p 都有 s[p]=s[p*i] [i=1,2..n/p]  n<=1000.


对素数p1,p2 若lcm(p1,p2)=p1*p2 <=n 那么

s[p1]=s[i*p1]=..=s[p1*p2] = s[p2]=s[i*p2] = 某个字符c.

质数p,q满足p*q<=n 那么p*2，q*2<=n
若质数p*2>n 那么位置p可以放任意一个字符.
若质数p*2<=n 那么这些位置都会等于同一个字符. 

所以标记这些放相同字符的位置 然后看出现次数最多的字符能否放入即可.

#include <bits/stdc++.h>
using namespace std;
const int N=1e3+5;
char s[N];
int n,f[N],mk[N],cnt[N];
vector<int> p;
void init()
{
	for(int i=2;i<N;i++)if(!mk[i]){
		p.push_back(i);
		for(int j=i+i;j<N;j+=i)	mk[j]=true;				
	}	
}
int main()
{
	scanf("%s",s+1);
	n=strlen(s+1);
	init();
	memset(mk,0,sizeof(mk));
	for(int i=0;i<p.size();i++)
	{
		if(p[i]*2>n)
			break;
		for(int j=1;j*p[i]<=n;j++)
			mk[p[i]*j]=true;
	}
	int tot=0,mx=0,pos;
	for(int i=1;i<=n;i++)
	{
		cnt[s[i]-'a']++;
		if(mk[i])
			tot++;
	}
	for(int i=0;i<26;i++)
		if(mx<cnt[i])
			mx=cnt[i],pos=i;
	if(mx<tot)
	{
		cout<<"NO"<<'\n';
		return 0;
	}
	puts("YES");
	for(int i=1;i<=n;i++)
	{
		if(mk[i])
			printf("%c",'a'+pos),cnt[pos]--,tot--;
		else
			for(int j=0;j<26;j++)
				if(j==pos&&cnt[j]>tot || (j!=pos&&cnt[j]))
				{
					printf("%c",'a'+j);
					cnt[j]--;
					break;
				}
	}
	return 0;
}