1. 程式人生 > >[BZOJ1257][CQOI2007]余數之和sum 數學+分塊

[BZOJ1257][CQOI2007]余數之和sum 數學+分塊

clu 取值 lld logs can family print 一個 line

題目鏈接:http://www.lydsy.com/JudgeOnline/problem.php?id=1257

題目所求為$$Ans=\sum_{i=1}^nk%i$$

將其簡單變形一下$$Ans=\sum_{i=1}^nk-\lfloor\frac{k}{i}\rfloor*i$$

$$Ans=n*k-\sum_{i=1}^{min(n,k)}\lfloor\frac{k}{i}\rfloor*i$$

容易知道$\frac{k}{i}$一共有$\sqrt{k}$種取值,可以利用分塊技巧。然後$\frac{k}{i}$的值相同的這一段區間內,$i$是一個等差數列,可以用等差數列求和$O(1)$計算整個區間的值。

 1 #include<cstdio>
 2 #include<cstring>
 3 #include<algorithm>
 4 using namespace std;
 5 typedef long long ll;
 6 int N,K;
 7 int main(){
 8     scanf("%d%d",&N,&K);
 9     ll ans=(ll)N*K,la,M=min(N,K);
10     for(int i=1;i<=M;i=la+1){
11         la=min(N,K/(K/i));
12 ans-=((la+i)*(la-i+1)>>1)*(K/i); 13 } 14 printf("%lld\n",ans); 15 return 0; 16 }

[BZOJ1257][CQOI2007]余數之和sum 數學+分塊