【AtCoder】ABC 418 E - Trapezium | 緑コーダーが解くAtCoder

atcoder.jp

配点: 100 点 / 実行時間制限: 2 sec / メモリ制限: 1024 MB / Difficulty: 12 / NoviSteps: 7Q

問題概要
制約
考察
実装例
コメント

問題概要

英小文字からなる長さ $N$ の文字列 $S$ が与えられるので、 $S$ がteaで終わる文字列であるかを判定せよ。

制約

$1 \leq N \leq 20$
$N$ は整数。

考察

$S$ の末尾の3文字を取り出すには、S.substr(N-3)とすればよい。

これがteaと一致していればYes、そうでなければNo。

もちろん、S[N-3], S[N-2], S[N-1]を取り出してそれぞれ比較しても OK 。

実装例

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

// ======================================== //

int main()
{
    int N;
    string S;
    cin >> N >> S;

    if (N < 3)
    {
        cout << "No" << endl;
        return 0;
    }

    if (S.substr(N - 3) == "tea")
        cout << "Yes" << endl;
    else
        cout << "No" << endl;

    return 0;
}