码迷,mamicode.com
首页 > 其他好文 > 详细

hdu 4658 Integer Partition

时间:2014-10-20 22:34:25      阅读:354      评论:0      收藏:0      [点我收藏+]

标签:style   blog   http   color   io   os   ar   for   sp   

http://acm.hdu.edu.cn/showproblem.php?pid=4658

这道题不会做,看着别人的写的。这道题求的是一个数的划分,但是划分中一个数不能重复k次,用到了五边形数定理,没看懂。

bubuko.com,布布扣
 1 #include <cstdio>
 2 #include <cstring>
 3 #include <algorithm>
 4 #define maxn 100005
 5 using namespace std;
 6 const int mod=1000000007;
 7 
 8 int dp[maxn];
 9 
10 void inti()//一个数的划分
11 {
12     dp[0]=1;
13     for(int i=1; i<=100000; i++)
14     {
15         dp[i]=0;
16         for(int j=1; ; j++)
17         {
18             int x=(3*j-1)*j/2;
19             if(x>i) break;
20             int s=dp[i-x];
21             if(x+j<=i) s=(s+dp[i-j-x])%mod;
22             if(j&1) dp[i]=(dp[i]+s)%mod;
23             else dp[i]=(dp[i]-s+mod)%mod;
24         }
25     }
26 }
27 
28 int main()
29 {
30     inti();
31     int t;
32     scanf("%d",&t);
33     while(t--)
34     {
35         int n,k;
36         scanf("%d%d",&n,&k);
37         int ans=dp[n];
38         for(int i=1; ; i++)
39         {
40             int x=k*i*(3*i-1)/2;
41             if(x>n) break;
42             int c=dp[n-x];
43             if(x+i*k<=n) c=(c+dp[n-x-k*i])%mod;
44             if(i&1) ans=(ans-c+mod)%mod;
45             else ans=(ans+c)%mod;
46         }
47         printf("%d\n",ans);
48     }
49     return 0;
50 }
View Code

 

hdu 4658 Integer Partition

标签:style   blog   http   color   io   os   ar   for   sp   

原文地址:http://www.cnblogs.com/fanminghui/p/4038696.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!