UOJ 021题解

原创于 2020-07-29 09:46:44 发布 · 216 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

本文探讨了一种算法优化策略，目标是最小化给定序列的特定操作，通过整除分块技巧来减少计算复杂度，实现O(nlog_2n)的时间效率。文章详细解析了数学推导过程，并提供了C++实现代码。

Solution

显然，题目要让我们最小化:

$k\sum_{i=1}^n \lfloor \frac {a_i} k \rfloor+\sum_{i=1}^n a_i\ mod\ k$

$=∑i=1n⌊aik⌋+∑i=1n(ai−k(⌊aik⌋))=\sum_{i=1}^n \lfloor \frac {a_i} k \rfloor+\sum_{i=1}^n (a_i-k(\lfloor \frac {a_i} k \rfloor))$

$=∑i=1n⌊aik⌋+∑i=1nai−k∑i=1n(⌊aik⌋)=\sum_{i=1}^n \lfloor \frac {a_i} k \rfloor+\sum_{i=1}^n a_i-k\sum_{i=1}^n(\lfloor \frac {a_i} k \rfloor)$

$=∑i=1nai−(k−1)∑i=1n⌊aik⌋=\sum_{i=1}^n a_i-(k-1) \sum_{i=1}^n \lfloor \frac {a_i} k \rfloor$

即，我们要让 $\sum_{i=1}^n \lfloor \frac {a_i} k \rfloor$ 越小越好。于是，我们可以暴力枚举 $k$ ，然后对这个式子用整除分块优化。

故时间复杂度为 $O(∑i=1n⌊ni⌋)O(\sum_{i=1}^n \lfloor \frac n i \rfloor)$ ，即 $O(nlog_2n)$ 。

注意本题轻微卡常，强制使用快读。

Code

#include <bits/stdc++.h>
#define int long long
#define rg register
using namespace std;
const int maxlen=1000000;

int n,tot=0,ans=1e15+7,maxv;
int a[2*maxlen+5],v[2*maxlen+5],pre[2*maxlen+5];

inline int read()
{
	int s=0,w=1;
	char ch=getchar();
	
	while (ch<'0'||ch>'9')
	{
		if (ch=='-')  w=-w;
		ch=getchar();
	}
	while (ch>='0'&&ch<='9')
	{
		s=(s<<1)+(s<<3)+(ch^'0');
		ch=getchar();
	}
	return s*w;
}

signed main()
{
	cin>>n;
	for (int i=1;i<=n;i++)
	{
		a[i]=read();
		v[a[i]]++;
		maxv=max(maxv,a[i]);
	}
	for (int i=1;i<=2*maxv;i++)  pre[i]=pre[i-1]+v[i];
	for (int i=1;i<=n;i++)  tot+=a[i];
	for (rg int k=1;k<=maxv;k++)
	{
		int now=0;
		for (rg int j=k;j<=maxv;j+=k)  now+=(pre[j+k-1]-pre[j-1])*(j/k);
		
		ans=min(ans,tot-(k-1)*now);
	}
	cout<<ans<<endl;
	
	return 0;
}