JOI'14 春1-2 愉快なロゴデザイン

問題文

https://www.ioi-jp.org/camp/2015/2015-sp-tasks/2015-sp-d1.pdf
ある日 K 理事長は,円状に ‘J’, ‘O’, ‘I’ の文字を並べたものをロゴとすることを思いついた.これには JOI を楽
しんで (enjoy) もらいたい
とする思いが込められている.

解法

同じ文字列を後ろにつなげて各文字累積和をとって始点決めて再帰で計算

ソースコード

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define FOR(i,n,m) for(int i=(n);i<(m);i++)
#define REP(i,n) FOR(i,0,n)
#define REPR(i,n) for(int i=(n);i>=0;i--)
#define all(vec) vec.begin(),vec.end()
using vi=vector<int>;
using vvi=vector<vi>;
using vl=vector<ll>;
using vvl=vector<vl>;
using P=pair<int,int>;
using PP=pair<int,P>;
using Pl=pair<ll,ll>;
using PPl=pair<ll,Pl>;
using vs=vector<string>;
#define fi first
#define se second
#define pb push_back
template<class T>bool chmax(T &a,const T &b){if(a<b){a=b;return true;}return false;}
template<class T>bool chmin(T &a,const T &b){if(a>b){a=b;return true;}return false;}
const ll MOD=1000000007LL;
const int INF=1<<30;
const ll LINF=1LL<<60;
int n;
string st;
vvi sum;
string joi("JOI");
int f(int x,int level){
    if(!level){
        return 0;
    }
    //xから始まるlevel列のコスト
    int len=1<<((level-1)*2);
    //J[x,x+len-1]
    int p=len-sum[x+len-1][0];
    if(x){
        p+=sum[x-1][0];
    }
    //O[x+len,x+len*2-1]
    int q=len-sum[x+len*2-1][1]+sum[x+len-1][1];
    int r=len-sum[x+len*3-1][2]+sum[x+len*2-1][2];
    return p+q+r+f(x+len*3,level-1);
}
int main(){
    cin>>n;
    int len=1<<(n*2+1);
    cin>>st;
    st+=st;
    sum.resize(len,vi(3));
    REP(i,len){
        REP(j,3){
            if(st[i]==joi[j]){
                sum[i][j]=1;
            }
        }
    }
    REP(i,len-1){
        REP(j,3){
            sum[i+1][j]+=sum[i][j];
        }
    }
    int ans=INF;
    REP(i,len/2){
        chmin(ans,f(i,n));
    }
    cout<<ans<<endl;
    return 0;
}