洛谷-哈希算法-P3405- Cities and States S

[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$。

题解

每个州和城市只要前两个字符对应就可以判断，所以我们每次只要对前两个字母进行哈希就可以了，

int Stringhash(string s){
    int code=0,x=26;
    for(int i=0;i<2;i++){
        code=code*x+(s[i]-'A');
    }
    return code;
}

因为字母是26进制，所以 x=26；

在将城市和州进行hash后，首先是第一种思路

void fun1(){
    int n,x,y,ans;
    cin>>n;
    vector<int> a;
    string s,c;
    for(int i=0;i<n;i++){
        cin>>c>>s;
        x=Stringhash(c);
        y=Stringhash(s);
        if(x==y) continue;
        a.push_back(x+y*26*26);
        for(int j=0;j<a.size()-1;j++){
            if(x*26*26+y==a[j]) ans++; 
        }
    }
    cout<<ans<<endl;

}

这种方法最有疑惑的点应该是 x+y * 26 * 26和x * 26 * 26+y
我们举一个例子
两个十进制数 12 和 34，如果我们要让两个二位十进制数组成一个数字1234，应该是12100+34=1234，如果是3位十进制数123 和 456组合就是1231000+456=123456，所以我们的两位26进制数组合在一起就是 x+y * 26 * 26
我们用动态数组存取，然后将两个数反过来组合找到是否有对应的数，最后输出ans。
但是这种方法我们用了两层for，时间复杂度是 O(n^2) ，容易超时

所以我们用map的思想，用数组来表示

void fun2(){
    int n,x,y,ans,a[700][700];
    cin>>n;
    vector<int> a;
    string s,c;
    for(int i=0;i<n;i++){
        cin>>c>>s;
        x=Stringhash(c);
        y=Stringhash(s);
        if(x==y) continue;
        a[x][y]++;
        ans+=a[y][x];
    }
    cout<<ans<<endl;

}

这样我们用 x代表城市，y代表州，每次有一个组合就标记一下，然后看相反的组合是否存在，加到ans上输出即可。

#include<bits/stdc++.h>
using namespace std;

int Stringhash(string s){
    int code=0,x=26;
    for(int i=0;i<2;i++){
        code=code*x+(s[i]-'A');
    }
    return code;
}


void fun1(){
    int n,x,y,ans;
    cin>>n;
    vector<int> a;
    string s,c;
    for(int i=0;i<n;i++){
        cin>>c>>s;
        x=Stringhash(c);
        y=Stringhash(s);
        if(x==y) continue;
        a.push_back(x+y*26*26);
        for(int j=0;j<a.size()-1;j++){
            if(x*26*26+y==a[j]) ans++; 
        }
    }
    cout<<ans<<endl;

}

void fun2(){
    int n,x,y,ans,a[700][700];
    cin>>n;
    vector<int> a;
    string s,c;
    for(int i=0;i<n;i++){
        cin>>c>>s;
        x=Stringhash(c);
        y=Stringhash(s);
        if(x==y) continue;
        a[x][y]++;
        ans+=a[y][x];
    }
    cout<<ans<<endl;

}



int main(){
    fun2();
    return 0;
}