首先,题目如下:
问题描述 一个间不容发的时刻:n个牛仔站立于一个环中,并且每个牛仔都用左轮手枪指着他旁边的人!每个牛仔指着他顺时针或者逆时针方向上的相邻的人。正如很多西部片那样,在这一刻,绳命是入刺的不可惜……对峙的场景每秒都在变化。每秒钟牛仔们都会分析局势,当一对相邻的牛仔发现他们正在互指的时候,就会转过身。一秒内每对这样的牛仔都会转身。所有的转身都同时在一瞬间发生。我们用字母来表示牛仔所指的方向。“A”表示顺时针方向,“B”表示逆时针方向。如此,一个仅含“A”“B”的字符串便用来表示这个由牛仔构成的环。这是由第一个指着顺时针方向的牛仔做出的记录。例如,牛仔环“ABBBABBBA”在一秒后会变成“BABBBABBA”;而牛仔环“BABBA”会变成“ABABB”。 这幅图说明了“BABBA”怎么变成“ABABB” 一秒过去了,现在用字符串s来表示牛仔们的排列。你的任务是求出一秒前有多少种可能的排列。如果某个排列中一个牛仔指向顺时针,而在另一个排列中他指向逆时针,那么这两个排列就是不同的。 输入格式 输入数据包括一个字符串s,它只含有“A”和“B”。 输出格式 输出你求出来的一秒前的可能排列数。 数据规模和约定 s的长度为3到100(包含3和100) 样例输入 BABBBABBA 样例输出 2 样例输入 ABABB 样例输出 2 样例输入 ABABAB 样例输出 4 样例说明 测试样例一中,可能的初始排列为:"ABBBABBAB"和 "ABBBABBBA"。 测试样例二中,可能的初始排列为:"AABBB"和"BABBA"。以上为题目,我的第一方法是使用暴力搜索,即列出它的所有初始排列,例如当前的排列为ABA,那么我先列出前一次所有的排序AAA,AAB,ABA,ABB,BAA,BAB,BBA,BBB。然后再按题目要求进行翻转,找出符合的排序。具体代码如下:
#include<iostream> #include"cstring" using namespace std; int length = 0; //记录数组的长度 char s[101]; //输入当前的排序 char pre[101]; //记录上一次的排序 char turn[101]; //记录翻转后的排序 int count = 0; //记录可能的次数 //按照题目要求进行翻转,如果成功则返回true bool success(){ if(turn[0]=='B'&&turn[length-1]=='A'){ turn[0]='A'; turn[length-1]='B'; for(int i=1;i<length-2;i++){ if(turn[i]=='A' && turn[i+1]=='B'){ turn[i]='B'; turn[++i]='A'; } } }else{ for(int i=0;i<length-1;i++){ if(turn[i]=='A' && turn[i+1]=='B'){ turn[i]='B'; turn[++i]='A'; } } } if(strcmp(s,turn)==0){ return true; }else{ return false; } } //用递归列出所有可能的排序 void f(int n){ if(n == length){ pre[n] = '\0'; strcpy(turn,pre); if(success()){ count++; } return; } pre[n] = 'A'; f(n+1); pre[n] = 'B'; f(n+1); } int main(){ cin>>s; length = strlen(s); f(0); cout<<count; return 0; }
当我提交到蓝桥杯官网时,运行超时,不过也很正常,因为用暴力算法时实在是太耗费时间了,思考了很久还是想不出改进方法后,只好从百度搜索,具体代码如下(对方是使用树形DP来做的,如果对此不熟悉,可以先百度一下):
#include<iostream> #include<cstring> using namespace std; char s[105]; int length; int t[105][2]; int chose(int start,int end) { memset(t,0,sizeof(t)); t[start][0] = 1; t[start][1] = 0; while(start != end) { start = (start+1)%length; if(s[start] - s[(start-1+length)%length] == 0) { t[start][0] = t[(start-1+length)%length][0] + t[(start-1+length)%length][1]; t[start][1] = 0; } else if(s[start] - s[(start-1+length)%length] < 0) { t[start][0] = t[(start-1+length)%length][0] + t[(start-1+length)%length][1]; t[start][1] = t[(start-2+length)%length][0] + t[(start-2+length)%length][1]; if(t[start][1] == 0) { t[start][1] = 1; } } else { t[start][0] = t[(start-1+length)%length][1]; t[start][1] = 0; } } return t[start][0] + t[start][1]; } int nochose(int p) { int start,end; int temp = 2; end = (p-1+length)%length; start = (p+2)%length; if(s[p] > s[(p-1+length)%length]) { if(s[(p-2+length)%length] - s[(p-1+length)%length] > 0) { end = (end-2+length)%length; temp += 2; } else { return 0; } } if(s[(p+2)%length] - s[(p+1)%length] > 0) { if(s[(p+3)%length] - s[(p+2)%length] < 0) { start += 2; start %= length; temp += 2; } else { return 0; } } if(temp >= length) { return 1; } return chose(start,end); } int main(){ int i; int sum; cin>>s; length = strlen(s); for(i=0;i<length;i++) { if(s[i] - s[(i+1)%length] > 0) { break; } } sum = 0; sum += chose((i+2)%length,(i-1+length)%length) + nochose(i); // cout<<"chose"<<endl; // cout<<chose((i+2)%length,(i-1+length)%length)<<endl<<nochose(i)<<endl<<sum<<endl; cout<<sum<<endl; return 0; }