頭良くなりたい人

文系大学生shadeのブログです。競技プログラミングや人文学の話題,受験ネタなど。

ABC053 B - A to Z String

問題はこちら atcoder.jp

方針

s に現れる最初のAから最後のZまでの文字列が,条件を満たす最長の文字列になります。

コード

#include <bits/stdc++.h>
using namespace std;

int main(){
    string s;
    cin>>s;

    int firstA=-1;
    int lastZ=-1;

    for(int i=0; i<s.size(); i++){
        if(s[i]=='Z'){
            lastZ=i;
        }

        if(firstA!=-1){
            continue;
        }

        if(s[i]=='A'){
            firstA=i;
        }
    }

    cout<<lastZ-firstA+1<<endl;
}