CPU算力分配 - 华为OD机试题

输入:
2 2
1 1
2 2
输出
1 2
说明
从A组中选出算力为1的CPU，与B组中算力为2的进行交换，使两组服务器的算力都等于3

输入:
2 2
1 2
2 3
输出
1 2

输入:
1 2
2
1 3
输出
2 3

输入:
3 2
1 2 5
2 4
输出:
5 4

模拟的题目

解题思路:

计算两组服务器的总算力。
根据总算力的差值，计算出需要交换的差值的一半。
遍历第一组服务器的算力，尝试在第二组服务器中找到合适的算力进行交换，使得两组服务器的总算力相等。
输出找到的交换方案。

import java.util.Arrays;
import java.util.HashSet;
import java.util.Scanner;
import java.util.Set;
import java.util.stream.IntStream;

/**
 * @author code5bug
 */
public class Main {
    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        int L1 = in.nextInt(), L2 = in.nextInt();
        // A 组服务器
        int[] a = IntStream.range(0, L1).map(i -> in.nextInt()).toArray();
        Arrays.sort(a);
        int sum1 = IntStream.of(a).sum();

        // B组服务器
        int sum2 = 0;
        Set<Integer> b = new HashSet();
        for (int i = 0; i < L2; i++) {
            int t = in.nextInt();
            b.add(t);
            sum2 += t;
        }

        // A 组服务器应该减少的数量
        int d = (sum1 - sum2) / 2;
        for (int x : a) {
            // B 组是否存在 target ，如果存在则和 target 交换即可 （交换后两组算力相等）
            int target = x - d;
            if (b.contains(target)) {
                System.out.println(String.format("%d %d", x, x - d));
                return;
            }
        }
    }
}