diff --git a/Chapter1_ArraysAndStrings/Q5_OneAway.cpp b/Chapter1_ArraysAndStrings/Q5_OneAway.cpp index 70aeb92..0aa6780 100644 --- a/Chapter1_ArraysAndStrings/Q5_OneAway.cpp +++ b/Chapter1_ArraysAndStrings/Q5_OneAway.cpp @@ -1,25 +1,38 @@ #include -#include +#include +#include using namespace std; class solution{ public: - bool oneAway(string s1, string s2){ - unordered_map m1; - unordered_map m2; - int diff = 0; - - for(auto i : s1) - m1[i]++; - for(auto i : s2) - m2[i]++; - - for(auto i : s1) - if(m1[i]!=m2[i]) - diff++; - - return diff<2?true:false; + bool oneAway(string str1, string str2){ + if(fabs(str1.size() - str2.size()) > 1) + return false; + string s1=str1.size()