LeetCode-859. 亲密字符串

给你两个字符串 s 和 goal ，只要我们可以通过交换 s 中的两个字母得到与 goal 相等的结果，就返回 true ；否则返回 false 。

交换字母的定义是：取两个下标 i 和 j （下标从 0 开始）且满足 i != j ，接着交换 s[i] 和 s[j] 处的字符。

例如，在 "abcd" 中交换下标 0 和下标 2 的元素可以生成 "cbad" 。
 

示例 1：

输入：s = "ab", goal = "ba"
输出：true
解释：你可以交换 s[0] = 'a' 和 s[1] = 'b' 生成 "ba"，此时 s 和 goal 相等。
示例 2：

输入：s = "ab", goal = "ab"
输出：false
解释：你只能交换 s[0] = 'a' 和 s[1] = 'b' 生成 "ba"，此时 s 和 goal 不相等。
示例 3：

输入：s = "aa", goal = "aa"
输出：true
解释：你可以交换 s[0] = 'a' 和 s[1] = 'a' 生成 "aa"，此时 s 和 goal 相等。
示例 4：

输入：s = "aaaaaaabc", goal = "aaaaaaacb"
输出：true
 

提示：

1 <= s.length, goal.length <= 2 * 104
s 和 goal 由小写英文字母组成

#include <iostream>
#include <string>
#include <unordered_map>
using namespace std;
class Solution {
public:
	bool buddyStrings(string s, string goal) {

		int count = 0;
		unordered_map<char, int> hashMp1;
		unordered_map<char, int> hashMp2;
		bool isBigThanTwo = false;

		if (s.length() != goal.length()) {
			return false;
		}

		for (int i = 0; i < s.size(); i++) {
			if (hashMp1.count(s[i]) == 0) {
				hashMp1[s[i]] = 1;
			}
			else {
				hashMp1[s[i]]++;
			}
			if (hashMp2.count(goal[i]) == 0) {
				hashMp2[goal[i]] = 1;
			}
			else {
				hashMp2[goal[i]]++;
			}

			if (s[i] != goal[i]) {
				count++;
			}
		}

		unordered_map<char, int>::iterator it = hashMp1.begin();
		while (it != hashMp1.end()) {
			if (hashMp1[it->first] > 1) {
				isBigThanTwo = true;
			}

			if (hashMp1[it->first] != hashMp2[it->first]) {
				return false;
			}
			it++;
		}

		if (isBigThanTwo && s == goal) {
			return true;
		}

		if (count == 2) {
			return true;
		}

		return false;
	}

};

int main() {

	Solution* ps = new Solution();

	cout << ps->buddyStrings("ab", "ba");

	return 0;
}