USACO备考冲刺必刷题 | P3405 Cities and States

学习C++从娃娃抓起！记录下USACO（美国信息学奥赛）备考学习过程中的题目，记录每一个瞬间。

附上汇总贴：USACO备考冲刺必刷题 | 汇总-CSDN博客

---

【题目描述】

Farmer John 有若干头奶牛。为了训练奶牛们的智力，Farmer John 在谷仓的墙上放了一张美国地图。地图上表明了每个城市及其所在州的代码（前两位大写字母）。

由于奶牛在谷仓里花了很多时间看这张地图，他们开始注意到一些奇怪的关系。例如，FLINT 的前两个字母就是 MIAMI 所在的 FL 州，MIAMI 的前两个字母则是 FLINT 所在的 MI 州。 确切地说，对于两个城市，它们的前两个字母互为对方所在州的名称。

我们称两个城市是一个一对「特殊」的城市，如果他们具有上面的特性，并且来自不同的省。对于总共 N 座城市，奶牛想知道有多少对「特殊」的城市存在。请帮助他们解决这个有趣的地理难题！

【输入】

输入共 N+1 行。

第一行一个正整数 N，表示地图上的城市的个数。 接下来 N 行，每行两个字符串，分别表示一个城市的名称（2∼10 个大写字母）和所在州的代码（2 个大写字母）。同一个州内不会有两个同名的城市。

【输出】

输出共一行一个整数，代表特殊的城市对数。

【输入样例】

6
MIAMI FL
DALLAS TX
FLINT MI
CLEMSON SC
BOSTON MA
ORLANDO FL

【输出样例】

1

【代码详解】

#include 
using namespace std;
int n, ans=0;
string s, s1, s2, a[200005], b[200005];
map m;
int main()
{
    cin >> n;  // 输入n
    for (int i=1; i<=n; i++) {  // 遍历n对字符串
        cin >> s >> s2;  // 输入城市和州
        s1 = s.substr(0, 2);  // 城市取前2个字符
        a[i] = s1+s2;  // 拼接成a[i]和b[i]
        b[i] = s2+s1;
        m[a[i]]++;  // 统计a[i]的个数
    }
    for (int i=1; i<=n; i++) {  // 所以所有字符串对
        if (a[i]==b[i]) continue;  // 因为同一个州不会有两个同名的城市，所以a[i]不能等于b[i]
        ans += m[a[i]]*m[b[i]];  // 用a[i]的个数乘以b[i]的个数
        m[a[i]] = 0;  // 乘完一次后，后面再遇到a[i]，不再重复计算
    }
    cout << ans << endl;
    return 0;
}

【运行结果】

6
MIAMI FL
DALLAS TX
FLINT MI
CLEMSON SC
BOSTON MA
ORLANDO FL
1