原題大賽。
一句話做法:取當前最優的比取當前劣的要更優。
點選檢視程式碼
/*
Tips:
你陣列開小了嗎?
你MLE了嗎?
你覺得是貪心,是不是該想想dp?
一個小時沒調出來,是不是該考慮換題?
打 cf 不要用 umap!!!
記住,rating 是身外之物。
該衝正解時衝正解!
Problem:
演算法:
思路:
*/
#include<bits/stdc++.h>
using namespace std;
//#define map unordered_map
#define forl(i,a,b) for(register long long i=a;i<=b;i++)
#define forr(i,a,b) for(register long long i=a;i>=b;i--)
#define forll(i,a,b,c) for(register long long i=a;i<=b;i+=c)
#define forrr(i,a,b,c) for(register long long i=a;i>=b;i-=c)
#define lc(x) x<<1
#define rc(x) x<<1|1
#define mid ((l+r)>>1)
#define cin(x) scanf("%lld",&x)
#define cout(x) printf("%lld",x)
#define lowbit(x) (x&-x)
#define pb push_back
#define pf push_front
#define IOS ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
#define endl '\n'
#define QwQ return 0;
#define ll long long
#define ull unsigned long long
#define lcm(x,y) x/__gcd(x,y)*y
#define Sum(x,y) 1ll*(x+y)*(y-x+1)/2
#define aty cout<<"Yes\n";
#define atn cout<<"No\n";
#define cfy cout<<"YES\n";
#define cfn cout<<"NO\n";
#define xxy cout<<"yes\n";
#define xxn cout<<"no\n";
#define printcf(x) x?cout<<"YES\n":cout<<"NO\n";
#define printat(x) x?cout<<"Yes\n":cout<<"No\n";
#define printxx(x) x?cout<<"yes\n":cout<<"no\n";
#define x first
#define y second
ll t;
ll n;
ll a[300010];
ll b[300010],ans;
priority_queue< array<ll,2>,vector<array<ll,2>>,greater<array<ll,2>> >q;
void solve()
{
cin>>n;
forl(i,1,n)
cin>>a[i],b[i]=1;
forl(i,1,n)
ans+=a[i],q.push({3*a[i],i});
forl(i,1,n-2)
{
//ll xx=q.top().x,yy=q.top().y;
auto [x,y]=q.top();
q.pop(),ans+=x,b[y]++;
if(b[y]!=n-1)
q.push({((b[y]+1)*(b[y]+1)-b[y]*b[y])*a[y],y});
}
cout<<ans<<endl;
}
int main()
{
IOS;
t=1;
// cin>>t;
while(t--)
solve();
/******************/
/*while(L<q[i].l) */
/* del(a[L++]);*/
/*while(L>q[i].l) */
/* add(a[--L]);*/
/*while(R<q[i].r) */
/* add(a[++R]);*/
/*while(R>q[i].r) */
/* del(a[R--]);*/
/******************/
QwQ;
}