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 Rust (judge verdict: Accepted)

use std::io;
use std::collections::HashMap;

fn main(){
   loop {
        let v = read_vu();
        let n = v[0];
        let m = v[1];

        if n == 0 && m == 0 {break;}

        let vh = make_acv(n);
        let vw = make_acv(m);

        let mh = make_hm(n, vh);
        let mw = make_hm(m, vw);

        let mut cnt = 0;
        
        for (hl, ch) in &mh {
            cnt += match mw.get(&hl) {
                Some(cw) => ch * cw,
                None => 0,
            };
        }
        println!("{}", cnt);
   }
}

fn read_vu() -> Vec<usize> {
   let mut buf = String::new();
   io::stdin().read_line(&mut buf).expect("read error");
   buf.trim().split_whitespace().map(|e| e.parse().unwrap()).collect()
}

fn make_acv(l:usize) -> Vec<u32> {
   let mut v: Vec<u32> = vec![0; l + 1];
   for i in 0 .. l {
       let mut buf = String::new();
       io::stdin().read_line(&mut buf).expect("read error");
       let x: u32 = buf.trim().parse().unwrap();
       v[i+1] = v[i] + x;
   }
   v
}

fn make_hm(l: usize, v:Vec<u32>) -> HashMap<u32, u32> {
   let mut map: HashMap<u32, u32> = HashMap::new();
   for i in 0..l {
       for j in i+1 .. l+1 {
           let d = v[j] - v[i];
           let cnt = map.entry(d).or_insert(0);
           *cnt += 1;
       }
   }
   map
}