Problem:
Square Route

スクウェア・ルート

English text is not available in this practice contest.

このたび新しい豪邸を建てることを決めた大富豪の品田氏は，どの街に新邸を建てようかと悩んでいる．実は，品田氏は正方形がとても好きという変わった人物であり，そのため少しでも正方形の多い街に住みたいと思っている．

品田氏は，碁盤目状に道路の整備された街の一覧を手に入れて，それぞれの街について，道路により形作られる正方形の個数を数えることにした．ところが，道と道の間隔は一定とは限らないため，手作業で正方形を数えるのは大変である．そこであなたには，碁盤目状の道路の情報が与えられたときに，正方形の個数を数えるプログラムを書いて欲しい．

Input

入力は複数のデータセットから構成されており，各データセットは以下のような構成になっている．

N M

h1

h2

...

hN

w1

w2

...

wM

1 行目には 2 つの正の整数 N, M (1 ≦ N, M ≦ 1500) が与えられる．続く N 行 h1, h2, ..., hN (1 ≦ hi ≦ 1000）は，道路と道路の南北方向の間隔を表す．ここで hi は北から i 番目の道路と北から i + 1 番目の道路の間隔である．同様に，続く M 行 w1, ..., wM (1 ≦ wi ≦ 1000）は，道路と道路の東西方向の間隔を表す．ここで wi は西から i 番目の道路と西から i + 1 番目の道路の間隔である．道路自体の幅は十分細いため考慮する必要はない．

図 D-1: 最初のデータセット

N = M = 0 は入力の終端を示しており，データセットには含めない．

Output

各データセットに対して，正方形の個数を 1 行に出力しなさい．たとえば，Sample Input の最初のデータセットにおいては，以下のとおり 6 個の正方形を含むので，このデータセットに対する出力は 6 となる．

図 D-2: 最初のデータセットに含まれる正方形

Sample Input

3 3
1
1
4
2
3
1
1 2
10
10
10
0 0

Output for the Sample Input

6
2

Solution in JAVA (judge verdict: Accepted)

import java.util.ArrayList;
import java.util.HashMap;
import java.util.Map;
import java.util.Scanner;

public class Main {

	static int[][] map;
	static int[][] directions8 = { { -1, -1 }, { -1, 0 }, { -1, 1 }, { 0, -1 }, { 0, 1 }, { 1, -1 }, { 1, 0 },
			{ 1, 1 } };
	static int[][] directions4 = { { -1, 0 }, { 1, 0 }, { 0, -1 }, { 0, 1 } };
	static int ans;

	public static void main(String[] args) {

		Scanner sc = new Scanner(System.in);

		while (true) {

			int N = sc.nextInt();
			int M = sc.nextInt();

			if ((N | M) == 0) {
				break;
			}
			ans = 0;
			int[] h = new int[N];
			int[] hSum = new int[N + 1];
			int[] w = new int[M];
			int[] wSum = new int[M + 1];

			hSum[0] = 0;
			for (int i = 0; i < N; i++) {
				h[i] = sc.nextInt();
				hSum[i + 1] = h[i] + hSum[i];
			}
			wSum[0] = 0;
			for (int i = 0; i < M; i++) {
				w[i] = sc.nextInt();
				wSum[i + 1] = w[i] + wSum[i];
			}

			Map<Integer, Integer> hMap = new HashMap<Integer, Integer>();
			for (int i = 0; i < N + 1; i++) {
				for (int j = i + 1; j < N + 1; j++) {
					int tmpLength = hSum[j] - hSum[i];
					if (hMap.containsKey(tmpLength)) {
						hMap.put(tmpLength, hMap.get(tmpLength) + 1);
					} else {
						hMap.put(tmpLength, 1);
					}
				}
			}

			Map<Integer, Integer> wMap = new HashMap<Integer, Integer>();
			for (int i = 0; i < M + 1; i++) {
				for (int j = i + 1; j < M + 1; j++) {
					int tmpLength = wSum[j] - wSum[i];
					if (wMap.containsKey(tmpLength)) {
						wMap.put(tmpLength, wMap.get(tmpLength) + 1);
					} else {
						wMap.put(tmpLength, 1);
					}
				}
			}

			for (int hLength : hMap.keySet()) {
				if (wMap.containsKey(hLength)) {
					ans += hMap.get(hLength) * wMap.get(hLength);
				}
			}

			System.out.println(ans);

		}

	}

	// BFS用に二つの配列を足し算する
	static int[] addArrayElms(int[] a, int[] b) {
		int[] c = new int[a.length];
		for (int i = 0; i < a.length; i++) {
			c[i] = a[i] + b[i];
		}
		return c;
	}

	// //二分探索
	// k <= num となる最小の配列要素kのインデックスを返す
	static private int binarySearch(long num, long[] orderedArray) {
		int lowerBorder = -1;
		int upperBorder = orderedArray.length;
		int mid;

		while (upperBorder - lowerBorder > 1) {
			mid = (upperBorder + lowerBorder) / 2;
			if (orderedArray[mid] <= num) {
				lowerBorder = mid;
			} else {
				upperBorder = mid;
			}
		}
		return lowerBorder;
	}

	// 二分探索
	// k <= num となる最小のList要素kのインデックスを返す
	static private int binarySearch(long num, ArrayList<Long> orderedList) {
		int lowerBorder = -1;
		int upperBorder = orderedList.size();
		int mid;

		while (upperBorder - lowerBorder > 1) {
			mid = (upperBorder + lowerBorder) / 2;
			if (orderedList.get(mid) <= num) {
				lowerBorder = mid;
			} else {
				upperBorder = mid;
			}
		}
		return lowerBorder;
	}

	// aとbの最小公倍数を求める
	public static int gcd(int a, int b) {
		return b == 0 ? a : gcd(b, a % b);
	}

	public static long gcd(long a, long b) {
		return b == 0 ? a : gcd(b, a % b);
	}
}