【NOIP模拟】愉快的logo设计

Description

K理事长正在思考日本信息学奥林匹克竞赛选手的应援道具的logo问题。某天，K理事长突发奇想，想要设计一个用’J’，’O’，’I’三种文字环形排列的logo，意为希望选手能从JOI中收获快乐的意思。
(注：“环形地”在日文中的表述为“円状に”，“円”读作“en”，再加上“JOI”三个字即为“enjoy”……)
如下所示，对于任意非负整数k，我们定义标号为k的JOI序列Sk为：
·S0为’J’，’O’，’I’中任一字符构成的长度为1的字符串
·S[k+1]为最初4^k个字符都是’J’，接下来的4^k个字符都是’O’，接下来的4^k个字符都是’I’，最后4^k个字符是字符串Sk的长为4^(k+1)的字符串
现在，K理事长在纸上写下了由4^K个文字构成的一个环形字符串，字符串中每个字符都是’J’，’O’，’I’中的一个。K理事长想要修改一些文字，使得得到的字符串从某个起点开始顺时针读一圈后可以得到SK。在满足条件的情况下，要求修改的文字数量最少。

Solution

这种有环的字符串问题，先把串在后面复制一份。
对于这种有递推公式的题目，我们先写一下递推公式
$S_k=JJJJ...J+OOOO...O+IIII...I+S_{k-1}$，JOI分别有4^{k-1}次方个
我们发现J、O、I从某位开始的数量是可以算的而且都是4的k次幂
那么我们给原串的J、O、I打一个前缀和，然后从第i位开始，逐个用应有的J、O、I的数量减去原串的数量，最后因为要加一个J或O或I，因为原串只含有这三个字符，所以直接加原串的字符就好了。然后与答案去一个min。
很水的一道题。

Code

#include<iostream>
#include<cstdio>
#include<cmath>
#include<algorithm>
#include<cstring>
#define fo(i,a,b) for(i=a;i<=b;i++)
#define fod(i,a,b) for(i=a;i>=b;i--)
const int maxn=1048580*2;
using namespace std;
int i,j,k,l,t,n,m,ans,yi,er,san;
char s[maxn*2];
int a[maxn][3];
int si[11]={1,4,16,64,256,1024,4096,16384,65536,262144,1048576};
int main(){
    freopen("logo.in","r",stdin);
    freopen("logo.out","w",stdout);
    scanf("%d",&k);
    scanf("%s",s+1);
    n=1;
    fo(i,1,k)n*=4;
    fo(i,1,n)s[i+n]=s[i];
    fod(i,2*n,1){
        a[i][0]=a[i+1][0];
        a[i][1]=a[i+1][1];
        a[i][2]=a[i+1][2];
        if(s[i]=='J'){
            a[i][0]++;
        }
        else if(s[i]=='O'){
            a[i][1]++;
        }
        else{
            a[i][2]++;
        }
    }
    ans=0x7fffffff;
    fo(i,1,n){
        yi=0;
        l=i;
        fod(j,k-1,0){
            yi+=(si[j]*3-(a[l][0]-a[l+si[j]][0])-(a[l+si[j]][1]-a[l+2*si[j]][1])-(a[l+2*si[j]][2]-a[l+3*si[j]][2]));
            l+=si[j]*3;
        }    
        ans=min(ans,yi);
    }
    printf("%d\n",ans);
}