题目链接
一道典型的贪心题目,可以考虑到的就是对所有人的期待座位排序,最后逐个进行计算,贪心进行到底,为了我自己偷懒用了c++的sort快排,用c的话也可以自己写一个快排。
另外,大佬用了一个特别巧妙的桶来做这道题,代码长度及难度骤降。
下面是我的做法:
#include<bits/stdc++.h>
using namespace std;
const int MAX_N = 100010;
int n,m,a[MAX_N],k[MAX_N];
int main()
{
scanf("%d%d",&n ,&m);
for(int i=1;i<=m;++i)scanf( "%d" ,&a[i]);
if(m>n){
printf("-1");
return 0;
}
sort(a+1,a+m+1);
int s=0;
long long ans=0;
for(int i=1;i<=m;++i){
s=max(s,a[i]);
while(k[s] && s<=n)s++;
if(s>n){
printf("-1");
return 0;
}
k[s]=1;
ans+=s-a[i];
s++;
}
printf("%1ld",ans);
return 0;
}
大佬的做法(修改成c的代码),最后的思路基本相同,只是这个桶直接就少了排序环节,以及判定环节:
#include<stdio.h>
#define max(x,y) (x)>(y)?(x):(y)
int sum[100010];
int main()
{
int n,m,i,x;
long long int ans=0;
scanf("%d%d",&n,&m);
if(m>n){
printf("-1\n");
return 0;
}
for(i=1;i<=m;i++){
scanf("%d",&x);
sum[x]++;
}
for(i=1;i<=n;i++){
x=max(0,sum[i]-1);
ans+=x;
sum[i+1]+=x;
}
printf("%lld\n",sum[n+1]?(-1):ans);
return 0;
}