[ARC 102C]Triangular Relationship

Atcoder ARC 102C

C:Triangular Relationship

题意：

你有两个数\(N,K\),找到有多少三元组\((a,b,c)\),满足其中元素为\([1,n]\)的正整数，同时\(a+b\),\(b+c\),\(c+a\)都是\(K\)的倍数,其中\(a\),\(b\),\(c\)互换顺序算不同方案。
\(N,K \le 200000\)

题解：

因为\(a+b \equiv b+c \equiv c+a \pmod K\),所以\(a \equiv b \equiv c \pmod K\) 或者 \(a \equiv b \equiv c \pmod {\frac{K}{2}}\)
随便统计一下就行了

过程：

1A

代码：

int n,k;
ll ans;
inline ll Pow(ll a,int b) {return a*a*a;}
signed main() {
	read(n); read(k);
	int t=n/k;
	if(k&1) ans=Pow(t,3);
	else {
		ans=Pow(t,3);
		t=t+(n%k>=k/2 ? 1 : 0);
		// printf("%d\n",t);
		ans+=Pow(t,3);
	}
	printf("%lld\n",ans);
	return 0;
}

用时：10min