【Lqb T336】Cowboys
【链接】 我是链接,点我呀:)
【题意】
【题解】
设 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());
}
}
}