题意:给出两个数n和k,统计(a,b,c)三元组满足(a+b)%k=0,(b+c)%k=0,(a+c)%k=0且1<=a,b,c<=n的数量。 n,k的范围是2*10^5
这题是一个学弟问我的。。可是数学并不是我的强项。。。然后百度了参考了这位大佬的题解 https://www.cnblogs.com/heyuhhh/p/10099416.html
非常感谢。但是感觉还是不是特别明白这几个式子为什么这么弄。。我的排列组合功底太差了。
AC代码:
import java.util.Scanner;
/*由(a+b)%k=0,(b+c)%k=0,(a+c)%k=0我们可以知道,a,b,c要么都为k的整数倍,要么当他们模上k时余数为k/2。
我们现在来考虑f(n,3),即从不大于n的数里面选出可重复的3个数满足条件的个数。
首先我们可以知道k的整数倍有cnt1 = n/k 个数,然后第二种情况有cnt2 = n/k+(n%k>=k/2)个数(前提是k为偶数)。
因为我们是从中选数,所以是n%k>=k/2。
对于第一种情况,不论k的奇偶都会贡献给答案。而第二种情况只有当k为偶数时才被贡献给答案。
统计结果时就分别考率选三个不同的丶选两个相同的以及三个都相同的情况,最后加起来就是了。
*/
public class Main {
public static void main(String[] args) {
Scanner sc=new Scanner(System.in);
int n=sc.nextInt();
int k=sc.nextInt();
long cnt1=n/k;
long cnt2=n/k+(n%k>=k/2?1:0);
long ans=cnt1*(cnt1-1)*(cnt1-2)+3*(cnt1-1)*cnt1+cnt1;
if(k%2==0)
ans+=cnt2*(cnt2-1)*(cnt2-2)+3*(cnt2-1)*cnt2+cnt2;
System.out.println(ans);
sc.close();
}
}