两个稀疏向量的点积

/* 
 * 题目思路： 
 * 1. 使用索引和值构造稀疏向量。 
 * 2. 使用Java Stream API进行遍历和计算。 
 */
import java.util.HashMap;
import java.util.Map;

class SparseVector {
    private Map<Integer, Integer> map;

    public SparseVector(int[] indices, int[] values) {
        map = new HashMap<>();
        for (int i = 0; i < indices.length; i++) {
            map.put(indices[i], values[i]);
        }
    }

    // 计算两个稀疏向量的点积
    public int dotProduct(SparseVector vec) {
        return map.entrySet().stream()
            .filter(entry -> vec.map.containsKey(entry.getKey()))
            .mapToInt(entry -> entry.getValue() * vec.map.get(entry.getKey()))
            .sum();
    }
}

// 使用示例
// SparseVector v1 = new SparseVector(new int[]{1, 3, 5}, new int[]{4, 2, 1});
// SparseVector v2 = new SparseVector(new int[]{3, 5, 6}, new int[]{1, 3, 2});
// int result = v1.dotProduct(v2);  // result is 5