题目:洛谷P3405。
题目大意:给你一些省市的名称(大写)和所在省的名称(两个大写字母),求有多少对城市满足:A城市的名字的前两个字母等于B城市所在省的名称,且A所在省的名称等于B城市的名字的前两个字母。
解题思路:很容易发现,对于每个城市,我们可以把它缩成4个字母,前两个字母为城市名字的前两个字母,后两个为所在省名。
然后发现,城市一共只有$26^4$种。
然后我们就可以把它看成一个四位26进制数,则用Hash转成十进制,然后用一个数组计数即可。
注意前两个字母不能完全等于后两个字母。
其他见代码。
时间复杂度$O(4n)$。
C++ Code:
#include<cstdio>
int n;
char s1[15],s2[15];
int h[456978];
inline int hash(char c1,char c2,char c3,char c4){return(((c1-'A')*26+c2-'A')*26+c3-'A')*26+c4-'A';}
int main(){
scanf("%d",&n);
int ans=0;
for(int i=1;i<=n;++i){
scanf("%s%s",s1,s2);
if(s1[0]!=s2[0]||s1[1]!=s2[1]){
int hh1=hash(s1[0],s1[1],s2[0],s2[1]),hh2=hash(s2[0],s2[1],s1[0],s1[1]);
ans+=h[hh2];
++h[hh1];
}
}
printf("%d\n",ans);
return 0;
}