题目:洛谷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; }