【bzoj1257】 余数之和sum 数论

    这道题目如果用暴力是过不了的,所以必须要优化。

    对于k mod i,我们将其转换为k-i*[k/i]([a]表示不大于a的最大整数),那么:

ans=Σ(i=1,n) (k mod i)=Σ(i=1,n) (k-i*[k/i])=n*k-Σ(i=1,n) (i*[k/i])

     我们发现,对于许多i(尤其是当i很大时),[k/i]的值都是相等的,事实上,如果设[k/i]=m,满足[k/i]=m的最小的i为u,那么满足条件的最大的i为v=[k/m]。简单的证明如下(写得不是很清楚,建议自己思考,可以无视):

     由于[k/u]=[k/v]=m,而v又为最大值,故有m*v<=k m*(v+1)>k,即k/m-1<v<=k/m,因此v正是[k/m]的值。

     由此,我们可以用等差数列求和来加速运算,对于所有相等的[k/i],可以很快地算出Σ(i=u,v) (i*[k/i])的值。由于k/i的值最多有sqrt(k)个,因此不会超时。代码中的i为上述的u,j为上述的v,详细的AC代码如下:

var
  n,k,i,j,ans:int64;
begin
  readln(n,k);
  i:=int64(1);
  ans:=n*k;
  while i<=n do
    begin
      if k div i=0 then break;
      j:=k div (k div i);
      if j>n then j:=n;
      ans:=ans-(i+j)*(j-i+int64(1))*(k div i) div int64(2);
      i:=j+int64(1);
    end;
  writeln(ans);
end.


2015.2.9

by lych

你可能感兴趣的:(数论)