【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());
        }
    }
}