题目链接:点击打开链接
题目大意:从n个数里面挑一个数当基数,然后再把剩余数里面不小于他的数字,变成他的倍数(如果不是的话就减小到能成为基数的倍数)然后相加,求最大的和。
题目思路:其实最容易想到的思路很简单啊,直接先sort排序一下,然后从第一个开始按照题目要求无脑算,但是很明显发现..时间复杂度O(n^2)..一看就是tle的节奏,所以就要改变想法。大家可以看,我先举个例子,比如我这个基数是4,然后后面一个数字是6一个是7,但是最后算和的时候,这俩都要当4算,发现了吧,在4~7中都按4,8~15都当8,这我们又想到了什么?只用算出相应范围内的数字的个数这道题就解决了!然而怎么算指定范围内的个数呢??毕竟每次基数不一样范围会变啊?这个时候就要使用前缀和了,先简单介绍一下,其实不要想的太难很简单的..算法竞赛宝典有相应的题目,有个最大子矩阵挺有意思,二维前缀和有空可以看一下..扯远了..回到这题,这题用的是一维前缀和。
前缀和:前缀和就是每个数组都记录从1到现在的所有个数的和。继续举例子..
比如a[1]=1,a[2]=2,a[3]=3,a[4]=4,a[5]=5,ok,接下来求前缀和,前
缀和核心代码就一条,就是图中这个,那很明显,
sum[1]=1,sum[2]=3,sum[3]=6,sum[4]=10,sum[5]=15,
你可能会问了,求这个有啥用呢?那我来提一个
问题,a3到a5的和是多少?按传统方法,a3+a4+a5,时间复杂度O(N),但是在前缀和面前,直接sum[5]-sum[2]不就好了嘛,O(1)美滋滋。
接下来就是要去重,vis标记一下就好了,接下来就是求每个范围内的数字个数,乘以当前范围对应的数字加一下就好了。记得加的时候后面要来个强制类型转换..单单temp和ans变long long还是会wa的..眼泪流下来.....
以下是代码:
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<vector>
#include<math.h>
#include<map>
using namespace std;
int a[200005],num[200005],vis[200005],n;
long long ans,temp;
int main(){
while(~scanf("%d",&n)){
ans=0;
memset(vis,0,sizeof(vis));
memset(num,0,sizeof(num));
for(int i=0;i<n;i++){
scanf("%d",&a[i]);
num[a[i]]++;
}
sort(a,a+n);
for(int i=1;i<=200000;i++){
num[i]+=num[i-1];
}
for(int i=0;i<n;i++){
if(!vis[a[i]]){
vis[a[i]]=1;
temp=0;
for(int j=1;j*a[i]<=a[n-1];j++){
int m=(j+1)*a[i]-1;
m=min(m,200000);
temp+=(long long)(num[m]-num[j*a[i]-1])*j*a[i];
}
ans=max(temp,ans);
}
}
printf("%lld\n",ans);
}
return 0;
}