Submission #59662162

---

Source Code Expand

#include <bits/stdc++.h>
#include<atcoder/all>
#define rep(i,n) for(int i=0;i<(n);i++)
using namespace std;
using namespace atcoder;
#define all(a) a.begin(),a.end()
typedef long long ll;
typedef pair<ll,ll> P;
constexpr ll mod=998244353;
typedef modint998244353 mi;

int substr_match(string s,string t){
    int n=s.size();
    string sr,sg,sb;
    rep(i,n){
        if(s[i]=='R'){
            sr.push_back('R');
            sg.push_back('B');
            sb.push_back('G');
        }
        if(s[i]=='G'){
            sr.push_back('B');
            sg.push_back('G');
            sb.push_back('R');
        }
        if(s[i]=='B'){
            sr.push_back('G');
            sg.push_back('R');
            sb.push_back('B');
        }
    }
    int ans=0;
    mi ht=0,hr=0,hg=0,hb=0;
    int base=1213;
    rep(i,n){
        hr=hr*base+sr[i];
        hg=hg*base+sg[i];
        hb=hb*base+sb[i];
        ht=ht+pow_mod(base,i,mod)*t[n-1-i];
        if(ht==hr||ht==hg||ht==hb){
            ans++;
        }
    }
    return ans;
}

int all_same(string s){
    rep(i,s.size()-1){
        if(s[i]!=s[i+1])return 0;
    }
    return 1;
}

int main(){
    int n;cin>>n;
    string s,t;cin>>s>>t;

    int ans=0;
    ans+=substr_match(s,t);
    ans+=substr_match(t,s);

    string diag;
    rep(i,n){
        if(s[i]==t[i])diag.push_back(s[i]);
        else{
            if(s[i]!='R'&&t[i]!='R')diag.push_back('R');
            if(s[i]!='G'&&t[i]!='G')diag.push_back('G');
            if(s[i]!='B'&&t[i]!='B')diag.push_back('B');
        }
    }
    ans-=all_same(diag);
    cout<<ans<<endl;
}

Submission Info

Compile Error

Main.cpp: In function ‘int all_same(std::string)’:
Main.cpp:3:31: warning: comparison of integer expressions of different signedness: ‘int’ and ‘std::__cxx11::basic_string<char>::size_type’ {aka ‘long unsigned int’} [-Wsign-compare]
    3 | #define rep(i,n) for(int i=0;i<(n);i++)
      |                               ^
Main.cpp:48:5: note: in expansion of macro ‘rep’
   48 |     rep(i,s.size()-1){
      |     ^~~

Judge Result