整数划分问题是算法中的一个经典命题之一。
所谓整数划分,是指把一个正整数n写成如下形式:
n=m1+m2+m3+....+mi;(其中mi为正整数,并且1<=mi<=n),则{m1,m2,m3,....,mi}为n的一个划分。
如果{m1,m2,m3,....,mi}中的最大值不超过m,即max{m1,m2,m3,....,mi} <= m,则称它属于n的一个m划分。这里我们记n的m划分的个数为f(n,m);
例如当n=4时,它有5个划分:{4}、{3,1}、{2,2}、{2,1,1}、{1,1,1,1};
注意:4=1+3和4=3+1被认为是同一个划分。
该问题是求出n的所有划分个数,即f(n,n)。下面我们考虑求f(n,m)的方法。
(一)方法一——递归法
根据n和m的关系,考虑下面几种情况:
(1)当n=1时,不论m的值为多少(m>0),只有一种划分,即{1};
(2)当m=1时,不论n的值为多少(n>0),只有一种划分,即{1,1,....1,1,1};
(3)当n=m时,根据划分中是否包含n,可以分为两种情况:
(a)划分中包含n的情况,只有一个,即{n};
(b)划分中不包含n的情况,这时划分中最大的数字也一定比n小,即n的所有(n-1)划分;
因此,f(n,n) = 1 + f(n, n - 1)。
(4)当n<m时,由于划分中不可能出现负数,因此就相当于f(n,n);
(5)当n>m时,根据划分中是否包含m,可以分为两种情况:
(a)划分中包含m的情况,即{m,{x1,x2,x3,...,xi}},其中{x1,x2,x3,...,xi}的和为n-m,可能再次出现m,因此是(n-m)的m划分,因此这种划分个数为f(n-m, m);
(b)划分中不包含m的情况,则划分中所有值都比m小,即n的(m-1)划分,个数为f(n, m - 1);
因此,f(n,m) = f(n - m,m) + f(n, m - 1)。
综合以上各种情况,可以看出,上面的结论具有递归定义的特征,其中(1)和(2)属于回归条件,(3)和(4)属于特殊情况,而情况(5)为通用情况,属于递归的方法,其本质主要是通过减少n或m以达到回归条件,从而解决问题。
其递归表达式如下所示。
(二)方法二——母函数
下面我们从另一个角度,即“母函数”的角度来考虑这个问题。
所谓母函数,即为关于x的一个多项式G(x):
有G(x) = a0 + a1*x + a2*x^2 + a3*x^3 + ......
则我们称G(x)为序列(a0, a1, a2,.....)的母函数。关于母函数的思路我们不做更过分析。
我们从整数划分考虑,假设n的某个划分中,1的出现个数记为a1,2的个数记为a2,.....,i的个数记为ai,
显然有:ak <= n/k(0<= k <=n)
因此n的划分数f(n,n),也就是从1到n这n个数字抽取这样的组合,每个数字理论上可以无限重复出现,即个数随意,使它们的综合为n。显然,数字i可以有如下可能,出现0次(即不出现),1次,2次,......,k次等等。把数字i用(x^i)表示,出现k次的数字i用(x^(i*k))表示,不出现用1表示。
例如,数字2用x^2表示,2个2用x^4表示,3个2用x^6表示,k个2用x^2k表示。
则对于从1到N的所有可能组合结果我们可以表示为:
G(x) = ( 1 + x + x^2 + x^3 + ... + x^n)*(1 + x^2 + x^4 + x^6 + ....)....(1 + x^n)
= g(x,1)*g(x,2)*g(x,3)*....*g(x,n)
= a0 + a1*x + a2*x^2 +...+ an*x^n + ....//展开式
上面的表达式中,每个括号内的多项式代表了数字i的参与到划分中的所有可能情况。因此,该多项式展开后,由于x^a *x^b = x^(a+b),因此x^i就代表了i的划分,展开后(x^i)项的系数也就是i的所有划分个数,即f(n,n) = an。
由此我们找到了关于整数划分的母函数G(x);剩下的问题就是,我们需要求出G(x)的展开后的所有系数。
为此,我们首先要做多项式乘法,对于我们来说,并不困难。我们把一个关于x的多项式用一个整数数组a[]表示,a[i]代表x^i的系数,即:
g(x) = a[0] + a[1]x + a[2]x^2 + ... + a[n]x^n;
例题:
Put Apples
Time Limit: 1000ms Memory limit: 65536K 有疑问?点这里^_^
题目描述
Your mission is to calculate the total number of different classifications; (Indicated by K) ;
5,1,1 and 1,5,1 are the same classifications;
输入
输出
示例输入
1 7 3
示例输出
8
该题代码:
1 #include<stdio.h> 2 int div(int n,int m) 3 { 4 if(n==1||m==1) 5 return 1; 6 else if(n==m) 7 return div(n,n-1)+1; 8 else if(n<m) 9 return div(n,n); 10 else return div(n,m-1)+div(n-m,m); 11 } 12 int main() 13 { 14 int n,k,l,m; 15 scanf("%d",&n); 16 while(n--) 17 { 18 scanf("%d %d",&m,&l); 19 k=div(m,l); 20 printf("%d\n",k); 21 } 22 return 0; 23 }
这是相对简单的分苹果问题,
模板代码实现如下:
1 #include <stdio.h> 2 3 #include <stdlib.h> 4 5 #include <string.h> 6 7 8 9 #define DEBUG 10 11 //递归法求解整数划分 12 13 unsigned long GetPartitionCount(int n, int max) 14 15 { 16 17 if(n == 1 || max == 1) 18 19 { 20 21 return 1; 22 23 } 24 25 if(n < max) 26 27 { 28 29 return GetPartitionCount(n, n); 30 31 } 32 33 if(n == max) 34 35 { 36 37 return 1 + GetPartitionCount(n, n - 1); 38 39 } 40 41 else 42 43 { 44 45 return GetPartitionCount(n - max, max) + GetPartitionCount(n, max - 1); 46 47 } 48 49 } 50 51 52 53 54 55 //母函数法求整数划分 56 57 58 59 #define MAXNUM 100 //最高次数 60 61 unsigned long a[MAXNUM]; 62 63 unsigned long b[MAXNUM]; 64 65 unsigned long c[MAXNUM]; //保存结果 66 67 68 69 //两个多项式进行乘法,系数分别保存在a和b中,结果保存到c,项的最大次数到MAXNUM 70 71 void Poly() 72 73 { 74 75 int i; 76 77 int j; 78 79 memset(c, 0, sizeof(c)); 80 81 for(i = 0; i < MAXNUM; i++) 82 83 { 84 85 for(j = 0; j < MAXNUM - i; j++) //j < MAXNUM - i,确保i+j不越界 86 87 { 88 89 c[i + j] += a[i] * b[j]; 90 91 } 92 93 } 94 95 } 96 97 //计算前N项的系数,即g(x,1)*g(x,2)*....*g(x,n)的展开结果 98 99 void Init(int m) 100 101 { 102 103 int i; 104 105 int j; 106 107 memset(a, 0, sizeof(a)); 108 109 memset(c, 0, sizeof(c)); 110 111 //第一个多项式:g(x) = x^0 + x^1 + x^2 + ... + x^n 112 113 for(i = 0; i < MAXNUM; i++) 114 115 { 116 117 a[i] = 1; 118 119 } 120 121 //for(j = 2; j <= MAXNUM; j++)//只能求f(n,n) 122 123 //通过修改这里,使得可以求f(n,m),对于任意的正整数n,m都适合 124 125 for(j = 2; j <= m; j++) 126 127 { 128 129 memset(b, 0, sizeof(b)); 130 131 //第i个多项式:g(x) = x^0 + x^i + x^(2k) + ... 132 133 for(i = 0; i <= MAXNUM; i += j) 134 135 { 136 137 b[i] = 1; 138 139 } 140 141 //多项式相乘:c = a * b 142 143 Poly(); 144 145 //将结果c保存到a中 146 147 memcpy(a, c, sizeof(c)); 148 149 } 150 151 } 152 153 154 155 //母函数方法得出整数划分相应的划分数目 156 157 //n:整数 158 159 //m:划分方法 160 161 void CalPrint(int n, int m) 162 163 { 164 165 if(n < m) 166 167 { 168 169 Init(n); 170 171 //由于n小于m,此时按n == m打印 172 173 printf("由于n小于m,所有(%d,%d) = (%d,%d) = %ldn", n, m, n, n, c[n]); 174 175 } 176 177 else 178 179 { 180 181 182 183 Init(m); 184 185 printf("整数划分(%d,%d)方法数目f(%d,%d) = %ldn", n, m, n, m, c[n]); 186 187 } 188 189 } 190 191 192 193 int main(int argc, char **argv) 194 195 { 196 197 int n; 198 199 int m; 200 201 unsigned long count; 202 203 printf("请输入要划分的整数:n"); 204 205 scanf("%d", &n); 206 207 printf("请输入划分数:n"); 208 209 scanf("%d", &m); 210 211 if(n <= 0) 212 213 { 214 215 fprintf(stderr, "输入的整数不能为非正数.n"); 216 217 return -1; 218 219 } 220 221 if(m <= 0) 222 223 { 224 225 fprintf(stderr, "输入的划分数不能为非正数.n"); 226 227 return -1; 228 229 } 230 231 count = GetPartitionCount(n, m); 232 233 234 235 printf("方法一:递归法n"); 236 237 printf("整数划分(%d,%d)的方法数为:%dnn", n, m, count); 238 239 240 241 printf("方法二:母函数法n"); 242 243 CalPrint(n,m); 244 245 246 247 #ifdef DEBUG 248 249 int i = 0; 250 251 for( i = 0; i < MAXNUM; i++) 252 253 { 254 255 printf("%9ld ", c[i]); 256 257 if((i + 1) % 10 == 0) 258 259 { 260 261 printf("n"); 262 263 } 264 265 } 266 267 printf("n"); 268 269 #endif 270 271 return 0;