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