【蓝桥杯省赛真题详解】排列小球（2018）

题目描述

输入输出样例
输入

3 6 0

输出

3

样例说明
用 r 表示红球，g 表示绿球，可能的方案包括：

rrrgggggg

grrrggggg

ggrrrgggg

参考代码

import java.util.Scanner;

public class Main {

	static int arr[] = new int[3];
	static int count = 0;

	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		int sum = 0;
		for (int i = 0; i < arr.length; i++) {
			arr[i] = sc.nextInt();
			sum += arr[i];
		}
		sc.close();
		f(sum, 0, -1);
		System.out.println(count);

	}

	private static void f(int sum, int x, int last) {
		if(sum==0) {
			count++;
			return;
		}
		for (int i = 0; i < arr.length; i++) {
			if(i==last)continue;
			for (int j = x + 1; j <= arr[i] && j <= sum; j++) {
				arr[i] = arr[i] - j;
				f(sum - j, j, i);
				arr[i] = arr[i] + j;
			}
		}

	}
}