折半枚举
对于许多题目,可能暴力搜索1 ~ n 是超时的,但是搜索1 ~ n/2 和n/2 ~ n就不会超时,此时我们可以考虑折半枚举的做法
题目:AtCoder - agc026_c
题意:给一个长为2n的字符串,问能找到从前往后读和从后往前读的这两个字符串相同的个数(从前往后的字符串与从后往前的字符串不能有公用)n<18
思路:由于n<18,所以2n会到36,直接暴力搜索会超时(试过了=.=);观察题目的第一个样例我们发现:我们把字符串从中间分开,前半部分红的和后半部分蓝的相同并且前半部分蓝的和后半部分红的相同时,才满足前后读时相同的。那么我们先枚举1~n,算出所有情况的数量,再枚举n-2n,找到符合条件的数量。
const int Len = 55;
int p[Len],v[Len],vis[Len],n;
ll res;
char s[Len];
map<pair<string,string>,int > mp;
void dfs1(int x,int ans,int cnt) {
int i;
if(ans == cnt + 1) {
memset(vis,0,sizeof vis);
string s1 = "",s2 = "";
for(i = 1 ; i <= cnt ; i ++) {
s1 += s[p[i]];
vis[p[i]] = 1;
}
for(i = n - 1 ; i >= 0 ; i --) {
if(!vis[i])s2 += s[i];
}
mp[make_pair(s1,s2)] ++;
return ;
}
for(i = x + 1 ; i < n ; i ++) {
if(!v[i]) {
v[i] = 1;
p[ans] = i;
dfs1(i,ans + 1,cnt);
v[i] = 0;
}
}
}
void dfs2(int x,int ans,int cnt) {
int i;
if(ans == cnt + 1) {
memset(vis,0,sizeof vis);
string s1 = "",s2 = "";
for(i = 1 ; i <= cnt ; i ++) {
s1 += s[p[i]];
vis[p[i]] = 1;
}
for(i = 2 * n - 1 ; i >= n ; i --) {
if(!vis[i])s2 += s[i];
}
res += mp[make_pair(s2,s1)];
return ;
}
for(i = x + 1 ; i < 2 * n ; i ++) {
if(!v[i]) {
v[i] = 1;
p[ans] = i;
dfs2(i,ans + 1,cnt);
v[i] = 0;
}
}
}
int main() {
int i,j,k;
cin >> n >> s;
for(i = 0 ; i <= n ; i ++) {
dfs1(-1,1,i);
}
for(i = 0 ; i <= n ; i ++) {
dfs2(n-1,1,i);
}
cout << res << endl;
return 0;
}
我们发现这种暴力搜索代码又长又臭,而且常数还大。我们发现,我们枚举的是1~n每一位取还是不取,所以可以用二进制枚举的方法,代码简洁明了
const int Len = 55;
int n;
char s[Len];
map<pair<string,string>,int> mp;
void sovle() {
string s1,s2;
int i,j;
ll res = 0;
for(i = 0 ; i <= (1 << n) - 1 ; i ++) {
s1 = "";s2 = "";
for(j = 0 ; j < n ; j ++) {
if(i & (1 << j))s1 += s[n - j - 1];//s[j]
else s2 += s[n - j - 1];
}
mp[make_pair(s1,s2)] ++;
}
for(i = 0 ; i <= (1 << n) - 1 ; i ++) {
s1 = "";s2 = "";
for(j = n - 1 ; j >= 0 ; j -- ) {
if(i & (1 << j ))s2 += s[2 * n - j - 1];//s[n + j]
else s1 += s[2 * n - j - 1];
}
res += mp[make_pair(s1,s2)];
}
cout << res << endl;
}
int main() {
cin >> n >> s;
sovle();
system("pause");
return 0;
}