【Lqb T336】Cowboys

【链接】 我是链接,点我呀:)
【题意】

出现AB就要交换一下
给你结果序列
问你原序列有多少种可能
首尾可以交换.

【题解】

设
dp[i][0]表示i和i-1不交换,达到前i个字符序列的状态的方案数
dp[i][1]表示i和i-1交换,达到前i个字符序列的状态的方案数
对于
s[i-1] s[i]
A B
B A
A A
B B
这四种情况分类讨论一下就好.
只要记着如果遇到了AB一定要交换就好
然后就不难想到转移方法
(转移方法在代码里看就好)

然后上面这种是首尾不交换的方案.
还需要考虑一种就是首尾可以交换的方案数.
这个时候,只需要把首部和尾部强行交换一下就好
然后在中间的n-2个数字再做一下相同的DP就好

方案数累加两次

【代码】

import java.io.*;
import java.util.*;

public class Main {
    
    
    static InputReader in;
    static PrintWriter out;
        
    public static void main(String[] args) throws IOException{
        //InputStream ins = new FileInputStream("E:\\rush.txt");
        InputStream ins = System.in;
        in = new InputReader(ins);
        out = new PrintWriter(System.out);
        //code start from here
        new Task().solve(in, out);
        out.close();
    }
    
    static int N = 1000;
    static class Task{
        String s;
        int n;
        int dp[][];
        
        public void solve(InputReader in,PrintWriter out) {
            dp = new int[N+10][2];
            s = in.next();
            n = s.length();
            dp[0][0] = 1;
            dp[1][0] = 1;
            for (int ii = 1;ii < n;ii++){
                int i = ii+1;
                if (s.charAt(ii-1)=='A' && s.charAt(ii)=='B') {
                    dp[i][0] = dp[i-1][1];
                    dp[i][1] = 0;
                }
                if (s.charAt(ii-1)=='B' && s.charAt(ii)=='A') {
                    dp[i][0] = dp[i-1][0] + dp[i-1][1];
                    dp[i][1] = dp[i-2][0] + dp[i-2][1];
                }
                if (s.charAt(ii-1)==s.charAt(ii)) {
                    dp[i][1] = 0;
                    dp[i][0] = dp[i-1][0]+dp[i-1][1];
                }
            }
            int ans = 0;
            ans = dp[n][0]+dp[n][1];
            if (s.charAt(n-1)=='B' && s.charAt(0)=='A') {
                StringBuilder sb = new StringBuilder(s);
                sb.setCharAt(n-1, 'A');
                sb.setCharAt(0, 'B');
                for (int i = 0;i <= n;i++)
                    for (int j = 0;j < 2;j++)
                        dp[i][j] = 0;
                dp[1][1] = 1;
                for (int ii = 1;ii <n-1;ii++) {
                    int i = ii+1;
                    if (s.charAt(ii-1)=='A' && s.charAt(ii)=='B') {
                        dp[i][0] = dp[i-1][1];
                        dp[i][1] = 0;
                    }
                    if (s.charAt(ii-1)=='B' && s.charAt(ii)=='A') {
                        dp[i][0] = dp[i-1][0] + dp[i-1][1];
                        dp[i][1] = dp[i-2][0] + dp[i-2][1];
                    }
                    if (s.charAt(ii-1)==s.charAt(ii)) {
                        dp[i][1] = 0;
                        dp[i][0] = dp[i-1][0]+dp[i-1][1];
                    }
                }
                ans = ans + dp[n-1][0] + dp[n-1][1];
            }
            out.println(ans);
        }
        
    }

    

    static class InputReader{
        public BufferedReader br;
        public StringTokenizer tokenizer;
        
        public InputReader(InputStream ins) {
            br = new BufferedReader(new InputStreamReader(ins));
            tokenizer = null;
        }
        
        public String next(){
            while (tokenizer==null || !tokenizer.hasMoreTokens()) {
                try {
                tokenizer = new StringTokenizer(br.readLine());
                }catch(IOException e) {
                    throw new RuntimeException(e);
                }
            }
            return tokenizer.nextToken();
        }
        
        public int nextInt() {
            return Integer.parseInt(next());
        }
    }
}