【UOJ21】[UR #1]缩进优化

最新推荐文章于 2020-11-27 19:31:57 发布

KikiDMW

最新推荐文章于 2020-11-27 19:31:57 发布

阅读量688

点赞数

分类专栏：技巧

本文链接：https://blog.csdn.net/KikiDMW/article/details/70143230

版权

技巧专栏收录该内容

9 篇文章 0 订阅

订阅专栏

题目链接
这里写图片描述

题解

http://vfleaking.blog.uoj.ac/blog/33
选择tab宽度为 $i$ ，能减少的代码量是 $(i-1)\sum a_j/i$ 。 $xi\dots x(i+1)-1$ 的所有值对前面的式子产生的贡献都是 $(i-1)x$ ，这样的话枚举每一个这样的区间，然后利用权值数组的前缀和快速求出区间内有多少数就好了。复杂度是 $O(nlogn)$ 。

#include<bits/stdc++.h>
using namespace std;

inline int read(){
    int x = 0, f = 1; char c = getchar();
    while(!isdigit(c)) { if(c == '-') f = -1; c = getchar(); }
    while(isdigit(c)) { x = x * 10 + c - '0'; c = getchar(); }
    return x * f;
}

const int N = 1000000 + 10;
typedef long long ll;
int n, a[N], s[N], mx;
ll sum, ans;

void init(){
    n = read();
    for(int i = 1; i <= n; i++){
        a[i] = read();
        s[a[i]]++;
        sum += a[i];
        mx = max(mx, a[i]);
    }
    for(int i = 1; i <= mx; i++) s[i] += s[i-1];
}

void work(){
    ll tmp;
    for(int i = 2; i <= mx; i++){
        tmp = 0;
        for(int j = i; j <= mx; j += i){
            if(j + i - 1 <= mx) tmp += (s[j+i-1] - s[j-1]) * (j / i);
            else tmp += (s[mx] - s[j-1]) * (j / i);
        }
        ans = max(ans, tmp * (i - 1));
    }
    printf("%lld\n", sum - ans);
}

int main(){
    init();
    work();
    return 0;
}