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

时间:2023-03-09 04:04:23
【UOJ#21】【UR#1】缩进优化

我好弱啊,什么题都做不出来QAQ

原题:

小O是一个热爱短代码的选手。在缩代码方面,他是一位身经百战的老手。世界各地的OJ上,很多题的最短解答排行榜都有他的身影。这令他感到十分愉悦。

最近,他突然发现,很多时候自己的程序明明看起来比别人的更短,实际代码量却更长。这令他感到很费解。经过一番研究,原来是因为他每一行的缩进都全是由空格组成的,大量的空格让代码量随之增大。

现在设小O有一份 n 行的代码,第 i 行有 ai 个空格作为缩进。

为解决这一问题,小O要给自己文本编辑器设定一个正整数的默认TAB宽度 x,然后对于每一行,编辑器从头至尾不断把连续 x 个空格替换成一个TAB,直到剩余空格数不足 x 个。

最终缩进所占代码量为空格数与TAB数的和。请你帮小O选择一个合适的 x,使得缩进所占代码量最小。

n,ai<=10^6

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

恩一个比较简单的思路是枚举塊的大小,然后暴力计算答案,令M为最大的aiO(nM)可以拿到20分

然后发现暴力计算的时候相同的ai可以合并起来,这样可以搞成O(M^2)的因为有两个点M比较小n比较大所以可以拿到40分

如果tab的长度是x那么没多换一个tab就减少x-1个空格,所以最终答案是n-(x-1)*Σai/x,接下来最大化(x-1)*Σai/x

还是枚举x,每次计算Σai/x,直接计算不太好算,就枚举ai/x的值,在输入ai后用sj记录ai==j的有多少个,枚举ai/x的值后就可以快速通过s计算个数再乘上ai/x这个值得到结果

有一个结论,Σ_{i=1}^{n}1/i=O(logn)

所以这个做法最终是O(MlogM)的,就可以ac辣

我好弱啊,NOI怎么玩嘛QAQ

代码:

 #include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
using namespace std;
#define ll long long
const ll inf=(ll)2e18;
int rd(){int z=,mk=; char ch=getchar();
while(ch<''||ch>''){if(ch=='-')mk=-; ch=getchar();}
while(ch>=''&&ch<=''){z=(z<<)+(z<<)+ch-''; ch=getchar();}
return z*mk;
}
int n,a[]; int mx=;
ll s[];
ll ans=;
int main(){freopen("ddd.in","r",stdin);
cin>>n;
for(int i=;i<=n;++i) a[i]=rd(),mx=max(mx,a[i]),++s[a[i]];
for(int i=;i<=mx;++i) s[i]+=s[i-];
for(int i=;i<=mx;++i){
int jjj=mx/i; ll bwl=;
for(int j=;j<=jjj;++j) bwl+=(s[(j+)*i->mx?mx:(j+)*i-]-s[j*i-])*j;
ans=max(ans,(i-)*bwl);
}
ll bwl=;
for(int i=;i<=n;++i) bwl+=a[i];
cout<<bwl-ans<<endl;
return ;
}