P3405 [USACO16DEC] Cities and States S

题目描述

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

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

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

输入格式

输入共 𝑁+1 行。

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

输出格式

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

代码:

package lanqiao;

import java.util.*;

public class Main{
    static int h[][];
    static int n,x,y;
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        n = sc.nextInt();
        int res = 0;
        h = new int[676][676];
        for(int i = 0;i < n;i ++)
        {
            String a = sc.next();
            String b = sc.next();
            x = (a.charAt(0) - 'A') * 26 + a.charAt(1) - 'A';
            y = (b.charAt(0) - 'A') * 26 + b.charAt(1) - 'A';
//            System.out.println(x + " " + y);
            if(x != y)
            {
                h[x][y] += 1;
            }
            res += h[y][x];
        }
        System.out.println(res);
    }
}