ZJUTOJ 1053 UCPRER 的繞口令

Description

Input
第一行輸入一個正整數 nn 表示兩個字符串的長度。

接下來兩行輸入兩個長度爲 nn 的字符串 aa，bb，且字符串中只有小寫英文字母。

數據範圍

Output
輸出一個整數，表示繞口令的繞口係數。

Sample Input 1
3
abb
baa
Sample Output 1
2

分析
其實大部分人第一反應就是暴力求解，使用雙層循環逐個判斷，但因爲n的範圍是10的5次方，使用循環時間複雜度就會到10的10次方，一定會超時。故換種思路，在a[i] != b[i] 的條件下，判斷之前出現過多少次 (b[i],a[i])即可。代碼如下：

#include<iostream>
#include<cstring>
using namespace std;
int main() {
	int n ,p,q;
	long long num = 0;
	int s[150][150] ;
	memset(s, 0, sizeof(s));
	string a, b;
	cin >> n;
	cin >> a >> b;
	for (int i = 0; i < n; i++) {
		if (a[i] == b[i]) {
			continue;
		}
		p = a[i] - '0';
		q = b[i] - '0';
		num += s[q][p];
		s[p][q]++;
	}
	cout << num;
	return 0;
}