CodeForces 731F VideoCards 前缀和+枚举

//http://codeforces.com/problemset/problem/731/F

#include<bits/stdc++.h>
using namespace std;
const int MAX=2e5+10;
typedef long long ll;
int N,A[MAX];
bool Used[MAX]={false};
ll Cnt[MAX]={0};
int main()
{
    cin>>N;
    int temp=0;
    for (int i=1;i<=N;i++)
    {
        cin>>A[i];
        temp=max(temp,A[i]);
        Cnt[A[i]]++;
    }
    sort(A+1,A+1+N);
    for (int i=1;i<=temp;i++)
        Cnt[i]+=Cnt[i-1];
    ll Ans=0;
    for (int i=1;i<=N;i++)
    {
        //枚举每个card当 leading video card 时的情况
        //其他的当作 secondary card 时必须减少到A[i]的倍数
        //如果A[j]在p*A[i]~(p+1)*A[i]-1之间 A[j]可以减少到p*A[i]来满足题目要求
        //利用前缀和思想 Cnt[i]记录1~i范围内的元素个数
        ll sum=0;
        if (!Used[A[i]]) //去掉这个优化会超时
            for (int x=A[i];x<=temp;x+=A[i])
                sum+=(Cnt[min(x+A[i]-1,temp)]-Cnt[x-1])*x,Used[A[i]]=true;
        Ans=max(Ans,sum);
    }
    cout<<Ans;
    return 0;
}