[USACO16DEC] Cities and States S
题目描述
Farmer John 有若干头奶牛。为了训练奶牛们的智力,Farmer John 在谷仓的墙上放了一张美国地图。地图上表明了每个城市及其所在州的代码(前两位大写字母)。
由于奶牛在谷仓里花了很多时间看这张地图,他们开始注意到一些奇怪的关系。例如,FLINT 的前两个字母就是 MIAMI 所在的 FL
州,MIAMI 的前两个字母则是 FLINT 所在的 MI
州。
确切地说,对于两个城市,它们的前两个字母互为对方所在州的名称。
我们称两个城市是一个一对「特殊」的城市,如果他们具有上面的特性,并且来自不同的州。对于总共 N N N 座城市,奶牛想知道有多少对「特殊」的城市存在。请帮助他们解决这个有趣的地理难题!
输入格式
输入共 N + 1 N + 1 N+1 行。
第一行一个正整数
N
N
N,表示地图上的城市的个数。
接下来
N
N
N 行,每行两个字符串,分别表示一个城市的名称(
2
∼
10
2 \sim 10
2∼10 个大写字母)和所在州的代码(
2
2
2 个大写字母)。同一个州内不会有两个同名的城市。
输出格式
输出共一行一个整数,代表特殊的城市对数。
样例 #1
样例输入 #1
6
MIAMI FL
DALLAS TX
FLINT MI
CLEMSON SC
BOSTON MA
ORLANDO FL
样例输出 #1
1
提示
数据规模与约定
对于 100 % 100\% 100% 的数据, 1 ≤ N ≤ 2 × 1 0 5 1 \leq N \leq 2 \times 10 ^ 5 1≤N≤2×105,城市名称长度不超过 10 10 10。
解题思路:
先把多个字母的a只取前两个字母,只有当前缀和州名不同的时候才处理。
1.st[a + b]++;将城市前缀 a 和州名 b 组合起来作为键,记录或更新该组合出现的次数。
2.ans += st[b + a];检查是否有反向组合 b + a 已经存在,若存在则将其计数加到 ans 中。
代码如下:
#include <iostream>
using namespace std;
#include <set>
#include <map>
#include <string>
#include <vector>
int n;
string a,b;
int ans=0;
map<string,int> st;
int main(){
cin >> n;
for(int i=1;i <= n;i++){
cin >> a >> b;
a = a.substr(0,2);
if(a != b) {
st[a+b] ++;
ans += st[b+a];
}
}
cout << ans;
return 0;
}