We use cookies to ensure you have the best browsing experience on our website. Please read our cookie policy for more information about how we use cookies.
  1. Prepare
  2. Algorithms
  3. Implementation
  4. Manasa and Stones
  5. Discussions

Manasa and Stones

  • wilson155079
     + 0 comments
    public static List<Integer> stones(int n, int a, int b) {
        int A = Math.min(a, b);
        int B = a + b - A;
        int val = A * (n - 1);

        // An - A - Ai + Bi -> A(n - 1) - i(A - B)
        return IntStream.range(0, n)
                .map(i -> val - (i * (A - B)))
                .distinct()
                .boxed()
                .collect(Collectors.toList());
}