逃离农场

题目：

牛牛在农场饲养了n只奶牛,依次编号为0到n-1, 牛牛的好朋友羊羊帮牛牛照看着农场.有一天羊羊看到农场中逃走了k只奶牛,但是他只会告诉牛牛逃走的k只奶牛的编号之和能被n整除。你现在需要帮牛牛计算有多少种不同的逃走的奶牛群。因为结果可能很大,输出结果对1,000,000,007取模。

例如n = 7 k = 4:
7只奶牛依次编号为0到6, 逃走了4只
编号和为7的有:{0, 1, 2, 4}
编号和为14的有:{0, 3, 5, 6}, {1, 2, 5, 6}, {1, 3, 4, 6},{2, 3, 4, 5}
4只牛的编号和不会大于18,所以输出5. 

输入描述:

输入包括一行,两个整数n和k(1 ≤ n ≤ 1000),(1 ≤ k ≤ 50),以空格分割。

输出描述:

输出一个整数表示题设所求的种数。

解析：本题可用动态规划，具体分为dp[i][k][t]为前i只奶牛中选k只，模为t的情况，该情况可由以下两种情况组成：

前i-1头奶牛中选取k只，不取第i头奶牛：dp[i-1][k][t]

前i-1头奶牛选取k-1只奶牛，即第i头奶牛必选，第i头奶牛编号为i-1，则若要选取第i头奶牛后模为t，则现在模必须为t+i-1<0?t+i-1+n:t+i-1： dp[i -1][k-1][(t+i-1)%n]

所以动态规划表达式为  dp[i][j][t] = dp[i-1][j][t] + dp[i-1][k-1][(t+i-1+n)%n]

import java.util.Scanner;


public class Main {
	public static void main(String[] args) {
		int N = 0;
		int K = 0;
		Scanner scanner = new Scanner(System.in);
		while (scanner.hasNextInt()) {
			N = scanner.nextInt();
			K = scanner.nextInt();
			Solution(N, K);
		}
	}
	private static void Solution(int n, int k) {
		long[][] dp = new long[55][1000];//不适用三维数组是因为内存空间超出限制
			dp[0][0] = 1;
			
		for (int i = 1; i <= n; i++) {
			for (int j = k; j >= 1; j--) {//从k到1是因为使用的是二维数组，若从小到大，值会被覆盖，计算错误，
				for (int x = 0; x < n; x++) {
					System.out.print( "dp["+(i-1)+"][" + j + "][" + x + "]=" + dp[j][x]+ ",dp["+(i-1)+"][" + (j - 1) + "]["	+ (x - i < 0 ? x - i + n : x - i) + "]="	+ dp[j - 1][x - i < 0 ? x - i + n : x - i]);
					dp[j][x] = (dp[j][x] + dp[j - 1][x - i < 0 ? x - i + n : x- i]);
					System.out.println("表达式为：dp["+(i)+"][" + j + "][" + x + "]=dp["+(i-1)+"][" + j+ "][" + x + "]+dp["+(i-1)+"][" + (j - 1) + "]["+ (x - i < 0 ? x - i + n : x - i) + "]=" + dp[j][x]);
				}
				System.out.println();
			}
			System.out.println("=====i=" + i + "=====");
		}
		System.out.println(dp[k][0]);
	}


	private static void Solution2(int n, int k) {
		int[][][] dp = new int[1000][55][1005];//不适用三维数组是因为内存空间超出限制
		for(int i=0;i<=n;i++){
			dp[i][0][0] = 1;
		}
		
		for (int i = 1; i <= n; i++) {
			for (int j = k; j >= 1; j--) {//从k到1是因为使用的是二维数组，若从小到大，值会被覆盖，计算错误，
				for (int x = 0; x < n; x++) {
					System.out.print( "dp["+(i-1)+"][" + j + "][" + x + "]=" + dp[i-1][j][x]+ ",dp["+(i-1)+"][" + (j - 1) + "]["	+ (x - i < 0 ? x - i + n : x - i) + "]="	+ dp[i-1][j - 1][x - i < 0 ? x - i + n : x - i]);
					dp[i][j][x] = (dp[i-1][j][x] + dp[i-1][j - 1][x - i < 0 ? x - i + n : x- i]);
					System.out.println("表达式为：dp["+(i)+"][" + j + "][" + x + "]=dp["+(i-1)+"][" + j+ "][" + x + "]+dp["+(i-1)+"][" + (j - 1) + "]["+ (x - i < 0 ? x - i + n : x - i) + "]=" + dp[i][j][x]);
				}
				System.out.println();
			}
			System.out.println("=====i=" + i + "=====");
		}
		System.out.println(dp[n][k][0]);
	}
}