UOJ #21. 【UR #1】缩进优化

【题解】

　　我们先从1到Max ai枚举Tab占几个空格，然后跳着统计答案。

　　跳着统计答案的意思是对于每个长度i，在值域上%i的循环节是i，所以我们把值域分成m/i个区间，一个个计算答案。这样复杂度是调和级数，也就是n*ln n的。

　　统计答案的时候可以用值域上的前缀和辅助。

#include<cstdio>
#include<cstring>
#include<algorithm>
#define LL long long
#define rg register
#define N 1000010
using namespace std;
int n,mx;
LL ans=1e17,a[N],sum[N],cnt[N];
inline int read(){
    int k=0,f=1; char c=getchar();
    while(c<'0'||c>'9')c=='-'&&(f=-1),c=getchar();
    while('0'<=c&&c<='9')k=k*10+c-'0',c=getchar();
    return k*f;
} 
int main(){
//    freopen("best.in","r",stdin);
//    freopen("best.out","w",stdout);
    n=read();
    for(rg int i=1;i<=n;i++){
        int x=read(); sum[x]+=x; cnt[x]++; mx=max(mx,x);    
    }
    for(rg int i=1;i<=mx;i++) sum[i]+=sum[i-1],cnt[i]+=cnt[i-1];
    for(rg int i=1;i<=mx;i++){
        LL tmp=0;
        for(rg int j=0;j<=mx;j+=i){
            int ed=min(j+i-1,mx);
            LL div=(cnt[ed]-cnt[max(0,j-1)])*(j/i);
            tmp+=div+(sum[ed]-sum[max(0,j-1)]-i*div);
        }
//        printf("%d %lld\n",i,tmp);
        ans=min(ans,tmp);
    }
    printf("%lld\n",ans);
    return 0;
}