Description
WCB某天买了非常多的糖果并把它们分成N份,依次分别有1,2,3,…,N个糖果。他想拿出其中的3份分给他的室友, 为了不让室友们闹意见,必须让这三份的糖果总数恰好能被三人均分。请问他一共有多少种不同的组合方案数?
Input
有多组输入数据,每组输入非负整数N(3≤N≤106),如果N=0,表示输入结束,这个样例不需要处理。
Output
每组数据输出一个整数独占一行,表示共有多少种方案,由于可能会很大,最后结果对109+7取模。
Sample Input
3
4
5
0
Sample Output
1
2
4
解题思路:
首先把N均分为三等份x,y,z,将N分为模3=1的数x、3个模3=2的数y、3个模3=0的数z。然后判断N模3余几,若余2则y和z都加一;若余1则y加1;若余0则不需要操作,因为如果余0,在开始的均分三等份时就已把xyz分好。例如,当N=7时,首先x=y=z=N/3=2(xyz都为整型),然后N模3余1,所以y=y+1=3;当N=8时;首先x=y=z=N/3=2,N模3余2,所以y和z都加1,y=3,z=3。
然后按取出的三份糖果来自哪里分四种情况讨论:
取自哪里 | 方案数 |
---|---|
三份都取自x | x*(x-1)*(x-2)/6 |
三份都取自y | y*(y-1)*(y-2)/6 |
三份都取自z | z*(z-1)*(z-2)/6 |
从xyz中各取一份 | x * y * z |
若三份都取自x,三份相加显然能被3整除;若三份都取自y,三份相加也为3的倍数,因为y为模3余1的数,所以3个y相加则为3的倍数+1+1+1也能被3整除;若三份都取自z,3个z相加则为3的倍数+2+2+2也能被3整除;若三份分别取自x、y、z则为三的倍速+0+1+2也能被3整除。
#include<stdio.h>
int mod=1000000007;
int main()
{
__int64 x,y,z;
int n,sum=0;
while(scanf("%d",&n)&&n!=0)
{
x=n/3; //膜3为0的个数
y=n/3; //膜3为1的个数
z=n/3; //膜3为2的个数
if(n%3==2)
{
y++; z++;
}
else if(n%3==1)
y++;
sum=((x*(x-1)*(x-2)/6)%mod+(y*(y-1)*(y-2)/6)%mod+(z*(z-1)*(z-2)/6)%mod+x*y*z%mod)%mod;
printf("%d\n",sum);
}
return 0;
}