小猴编程周赛C++ | 2022年秋第5周T2双字母字符串

学习C++从娃娃抓起！记录下在学而思小猴编程学习过程中的题目，记录每一个瞬间。侵权即删，谢谢支持！

附上汇总贴：小猴编程C++ | 汇总-CSDN博客

---

【题目描述】
小猴正在挑战一个关于字符串的“简单题”：给定n个只包含大小写字母的字符串，所有字符串的长度均为2。要求在这n个字符串中找出任意两个字符串s和t，使得s和t在不区分大小写的情况下有且只有一个位置上的字母相同，请问这样成对的字符串一共有多少对。
例如，给定4个字符串“ab”，“CA”，“bA”，“bB”，那么符合要求的成对字符串有3对：“ab”和“bB”，“CA”和“bA”，“bA”和“bB”。注意这里的“ab”和“CA”是不满足要求的成对字符串，因为a和A不在同一位置上，即a在第1个位置上，而A在第2个位置上。
【输入】
第一行，包含一个整数t，表示共有t组测试数据；
每组测试数据：
第一行，包含一个整数n，表示有多少个字符串；
接下来n行，每行包含一个字符串$s_i$，保证字符串$s_i$长度为2且只包含大小 写字母。
【输出】
一行，包含一个整数，表示结果。
【输入样例】

2
4
kk
KK
aB
Ab
6
jf
JF
jk
jK
Jk
JK

【输出样例】

0
8

【代码详解】

#include <bits/stdc++.h>
using namespace std;
const int N = 1e5+10;
int cnt[26][26];
int main()
{
    int t, n;
    string s;
    cin >> t;
    while (t--) {
        cin >> n;
        memset(cnt, 0, sizeof(cnt));  // T组数据，每次都需初始化为0
        for (int i=1; i<=n; i++) {  // 输入n个字符串
            cin >> s;
            int x = s[0]-'a';  // 转为字母的相对位置
            int y = s[1]-'a';  
            if ('A'<=s[0] && s[0]<='Z') x = s[0]-'A';  // 对于大写的字母进行预处理
            if ('A'<=s[1] && s[1]<='Z') y = s[1]-'A';
            cnt[x][y]++;  // 使用箱排序的思想，记录x和y位置的字母出现的次数
        }
        long long ans = 0;
        for (int i=0; i<=25; i++) {  // 枚举
            for (int j=0; j<=25; j++) {
                if (cnt[i][j]!=0) {  // 如果x和y构成的字符串有出现
                    for (int k=0; k<=25; k++) {  // 再枚举
                        for (int c=0; c<=25; c++) {
                            if ((i!=k && j==c) || (i==k && j!=c)) {  // 找到满足条件，即有且只有1个位置上的字母相同
                                ans += cnt[i][j] * cnt[k][c];  // 结果为两者相乘（最后结果需要除以2）
                            }
                        }
                    }
                }
            }
        }
        printf("%lld\n", ans/2);
    }
    return 0;
}

【运行结果】

2
4
kk
KK
aB
Ab
0
6
jf
JF
jk
jK
Jk
JK
8