题目描述
n 个人从一楼乘电梯回家,给出他们想要到达的楼层,请计算所 有人乘梯所用时间之和。时间的计算方法:在一楼进电梯,第 1 个人 用时 1 秒,第 2 个人用时 2 秒,……,第 i 个人用时 i 秒;电梯每上升一层,用时 1 秒;出电梯时,在同一楼层,第 1 个人用时 1 秒,第 2 个人用时 2 秒,……,第 i 个人用时 i 秒。
输入格式
两行。第一行一个正整数 n,表示乘梯人数;第二行 n 正整数,表示 每个人要到达的楼层。
输出格式
一行。一个整数,表示整个用时,单位是秒。
输入样例:
3
13 18 13
输出样例:
27
代码:
#include<iostream>
#include<cmath>
#include<cstdio>
using namespace std;
int cnt[10000],ans;
int main(){
int n,maxn=0;
cin>>n;
for(int i=1;i<=n;i++){
int x;
cin>>x;
cnt[x]++;
maxn=max(maxn,x);
}
ans+=(1+n)*n/2;
for(int i=2;i<=maxn;i++){
ans++;
if(cnt[i]!=0){
ans+=(1+cnt[i])*cnt[i]/2;
}
}
cout<<ans;
return 0;
}