[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。
省脑子做法
#include<iostream>
#include<unordered_map>
using namespace std;
int res;
unordered_map<string,int> ct;
int main(){
int n;cin>>n;
while(n--){
string a,b;cin>>a>>b;
a=a.substr(0,2);
if(a==b)continue;
string c=a+b;
b=b+a;
ct[c]++;
if(ct.count(b))res+=ct[b];
}
cout<<res;
return 0;
}
没事找事做法
#include<iostream>
using namespace std;
using ULL=unsigned long long ;
const int N=3005033;
ULL p[5],h[N];
const int P=131;
int res;
int main(){
int n;cin>>n;
p[0]=1;
p[1]=131;
p[2]=p[1]*P;
for(int i=1;i<=n;i++){
string s1,s2,s3;cin>>s1>>s2;
s1=s1.substr(0,2);
if(s1==s2)continue;
s3=s1+s2;//城市+州
s1=s2+s1;//州+城市
ULL tt=0;
for(int i=0;i<s3.size();i++){
tt=tt*P+s3[i];
}
tt%=N;
h[tt]++;
tt=0;
for(int i=0;i<s1.size();i++){
tt=tt*P+s1[i];
}
tt%=N;
if(h[tt])res+=h[tt];
}
cout<<res;
return 0;
}