Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

How to split odd and even numbers and sum of both in a collection using Stream

How can I split odd and even numbers and sum both in a collection using stream methods of Java 8?

public class SplitAndSumOddEven {

    public static void main(String[] args) {

        // Read the input
        try (Scanner scanner = new Scanner(System.in)) {

            // Read the number of inputs needs to read.
            int length = scanner.nextInt();

            // Fillup the list of inputs
            List<Integer> inputList = new ArrayList<>();
            for (int i = 0; i < length; i++) {
                inputList.add(scanner.nextInt());
            }

            // TODO:: operate on inputs and produce output as output map
            Map<Boolean, Integer> oddAndEvenSums = inputList.stream(); // Here I want to split odd & even from that array and sum of both

            // Do not modify below code. Print output from list
            System.out.println(oddAndEvenSums);
        }
    }
}
like image 601
Bhaumik Thakkar Avatar asked Jan 22 '16 05:01

Bhaumik Thakkar


1 Answers

You can use Collectors.partitioningBy which does exactly what you want:

Map<Boolean, Integer> result = inputList.stream().collect(
       Collectors.partitioningBy(x -> x%2 == 0, Collectors.summingInt(Integer::intValue)));

The resulting map contains sum of even numbers in true key and sum of odd numbers in false key.

like image 81
Tagir Valeev Avatar answered Nov 08 '22 07:11

Tagir Valeev