Skip to content

1040 有几个 PAT

字符串 APPAPT 中包含了两个单词 PAT,其中第一个 PAT 是第 2 位(P),第 4 位(A),第 6 位(T);第二个 PAT 是第 3 位(P),第 4 位(A),第 6 位(T)。

现给定字符串,问一共可以形成多少个 PAT

输入格式:

输入只有一行,包含一个字符串,长度不超过 105,只包含 PAT 三种字母。

输出格式:

在一行中输出给定字符串中包含多少个 PAT。由于结果可能比较大,只输出对 1000000007 取余数的结果。

输入样例:

tex
APPAPT
APPAPT

输出样例:

tex
2
2

Solution:

java
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader in = new BufferedReader(new InputStreamReader(System.in));
        String s = in.readLine();
        int length = s.length();
        int p = 0, pa = 0, pat = 0;
        for (int i = 0; i < length; i++) {
            char c = s.charAt(i);
            if (c == 'P') {
                p++;
            } else if (c == 'A') {
                pa += p;
                pa %= 1000000007;
            } else if (c == 'T') {
                pat += pa;
                pat %= 1000000007;
            }
        }
        System.out.println(pat);
    }
}
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader in = new BufferedReader(new InputStreamReader(System.in));
        String s = in.readLine();
        int length = s.length();
        int p = 0, pa = 0, pat = 0;
        for (int i = 0; i < length; i++) {
            char c = s.charAt(i);
            if (c == 'P') {
                p++;
            } else if (c == 'A') {
                pa += p;
                pa %= 1000000007;
            } else if (c == 'T') {
                pat += pa;
                pat %= 1000000007;
            }
        }
        System.out.println(pat);
    }
}

Released under the MIT License.