P3405 [USACO16DEC] Cities and States S(哈希，map

[USACO16DEC] Cities and States S

题目描述

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

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

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

输入格式

输入共 $N+1$ 行。

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

输出格式

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

样例 #1

样例输入 #1

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

样例输出 #1

1

提示

数据规模与约定

对于 $100\%$ 的数据，$1\le N\le 2\times 10^5$，城市名称长度不超过 $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;
}