2017年校招全国统一模拟笔试(第四场)_逃离农场

[编程题] 逃离农场

时间限制：2秒

空间限制：32768K

牛牛在农场饲养了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),以空格分割。

输出描述:

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

输入例子1:

7 4

输出例子1:

5

 这种dp题目还不是很难，对于我这种acm打铁选手来说还能做，

 至于正规的acm比赛就呵呵了，言归正传。
 要保留三种状态，i 表示在前i头奶牛中选。
  j 表示在前i头奶牛中选了多少头。s 表示j头奶牛标号和的n的余数。
 构建数组 ll dp[1005][55][1005]。
 状态转移方程 dp[i][j][s]=dp[i-1][j-1][(s+n-num[i])%n]+dp[i-1][j][s];
 为什么保留余数，可以大大降低空间复杂度和时间复杂度。
 s-num[i]之前要先加n，防止负数出现。
最后还是超了空间，那就改成滚动数组，dp1和dp2。

ac代码

                #include<iostream>
		#include<cstdio>
		#include<cstring>
		#include<vector> 
		using namespace std;
		typedef long long ll;
		const ll mod=1e9+7;
		
		ll num[1005];
		vector<ll> dp1[55],dp2[55];
		int main()
{
               ll n,k,ans;
		for(ll i=0;i<=1000;i++)	
		    num[i]=i-1;
	    
	    while(scanf("%lld%lld",&n,&k)==2)
	    {
	    	for(ll i=0;i<55;i++) 
	    	{
	    		dp1[i].assign(1005,0);
	    		dp2[i].assign(1005,0);
			}
			
			dp1[0][0]=dp2[0][0]=1;      
	    	ans=0;
	    	
	    	for(ll i=1;i<=n;i++)
	        {
	    	    for(ll j=1;j<=k&&j<=i;j++)
	    		    for(ll s=0;s<n;s++)
					    dp2[j][s]=(dp1[j-1][(s+n-num[i])%n]+dp1[j][s])%mod;
					    
			    for(ll i=0;i<55;i++) 
	    		    dp1[i]=dp2[i];
		    }
		    printf("%lld\n",dp2[k][0]);
		}
		return 0;
}