Description
Today on a lecture about strings Gerald learned a new definition of string equivalency. Two strings a and b of equal length are called equivalent in one of the two cases:
- They are equal.
- If we split string a into two halves of the same size a1 and a2, and string b into two halves of the same size b1 and b2, then one of the following is correct:
- a1 is equivalent to b1, and a2 is equivalent to b2
- a1 is equivalent to b2, and a2 is equivalent to b1
As a home task, the teacher gave two strings to his students and asked to determine if they are equivalent.
Gerald has already completed this home task. Now it's your turn!
Input
The first two lines of the input contain two strings given by the teacher. Each of them has the length from 1 to 200 000 and consists of lowercase English letters. The strings have the same length.
Output
Print "YES" (without the quotes), if these two strings are equivalent, and "NO" (without the quotes) otherwise.
Sample Input
aaba abaa
YES
aabb abab
NO
Hint
In the first sample you should split the first string into strings "aa" and "ba", the second one — into strings "ab" and "aa". "aa" is equivalent to "aa"; "ab" is equivalent to "ba" as "ab" = "a" + "b", "ba" = "b" + "a".
In the second sample the first string can be splitted into strings "aa" and "bb", that are equivalent only to themselves. That's why string "aabb" is equivalent only to itself and to string "bbaa".
思路:DFS,注意当a1,a2,b1,b2长度为奇数时,若四者不满足条件则直接return false。
递归判断是要先判断a1==b2&&a2==b1再判断a1==b1&&a2==b2,因为这个无限TLE。。。一开始还以为是用了string超时....
AC代码
char []版:耗时15ms:
#include <iostream>
#include <cstring>
#include <string>
#include <string.h>
#include <stdio.h>
#include <algorithm>
using namespace std;
typedef long long LL;
#define max(a,b) (a>b?a:b)
#define min(a,b) (a<b?a:b)
char a[200001],b[200001];
bool find(char a[],char b[],int len)
{
if(!strncmp(a,b,len))return true;
if(len & 1) return false;
len>>=1;
if((find(a,b+len,len)&&find(a+len,b,len))||(find(a,b,len)&&find(a+len,b+len,len)))
return true;
return false;
}
int main()
{
scanf("%s%s",a,b);
int len = strlen(a);
if(find(a,b,len))
printf("YES\n");
else
printf("NO\n");
return 0;
}
string版,耗时592ms:
#include <iostream>
#include <cstring>
#include <string>
#include <string.h>
#include <cstdio>
#include <algorithm>
using namespace std;
typedef long long LL;
#define max(a,b) (a>b?a:b)
#define min(a,b) (a<b?a:b)
bool find(string a,string b,int len)
{
if(a==b)return true;
if(len&1)
return false;
string a1,a2,b1,b2;
len>>=1;
for(int i=0;i<len;++i)
{
a1+=a[i];
a2+=a[i+len];
b1+=b[i];
b2+=b[i+len];
}
if((find(a1,b2,len)&&find(a2,b1,len))||(find(a1,b1,len)&&find(a2,b2,len)))// || 前后的条件交换顺序就超时了....
return true;
return false;
}
int main()
{
string a,b;
while(cin>>a>>b)
{
int len=a.length();
if(find(a,b,len))
cout<<"YES"<<endl;
else
cout<<"NO"<<endl;
}
return 0;
}