题意:
Monocarp得到一份工作,每天要工作 m 分钟,他有一个爱好,喜欢在休息的时候喝咖啡,但是他的老板不乐意了,就给他规定了个
时间 d,在 d 分钟内只能喝一杯咖啡。
现给出Monocarp喝 n 杯咖啡的时间点,问最少需要几天喝完?
并输出每个时间点的咖啡在第几天喝。
样例1解释:
4 5 3
3 5 1 2
Monocarp有 n 杯咖啡需要喝,但老板规定,在 3 分钟内只能喝一杯,
那么在第一天Monocarp可以喝时间点为 1,5 的咖啡;
第二天可以喝时间点为 2 的咖啡;
第三天可以喝时间点为 3 的咖啡;
那么,最少需要三天才能喝完这 n 杯咖啡 ;
第一杯咖啡 3 在第三天喝,输出 3;
第二杯咖啡 5 在第一天喝,输出 1;
第三杯咖啡 1 在第一天喝,输出 1;
第四杯咖啡 2 在第二天喝,输出 2;
题解:
根据贪心的思想,对于每一天,尽可能的多喝,这样才会使中天数最少;
首先将咖啡按时间点从小到大排个序,然后找出第一天最多喝的咖啡,第二天最多和的咖啡,以此类推,
直到所有咖啡全部喝完。
那么该怎么找第一天最多和的咖啡呢?
这里我使用了一个 map 中的 lower_bound() 函数,首先将喝咖啡的 n 个时间点存在 map 的first中。根据 map 的特性,
他会自动按照 first值 由小到大排序。
由贪心思想可得,时间点最小(假设为 x)的咖啡要在第一天喝,然后,在 map 中二分查找时间点 ≥ x+d+1 的最小的时刻(假设为 y),
那么 y 时刻的咖啡也要在第一天喝,接着查找时间点 ≥ y+d+1 的最小时刻,以此类推,直到搜索完所有满足条件的时间点,那么第一天的
喝咖啡的时间点就搜索完毕了,并且,在确定某一时间点在第一天喝后,要将其从 map 数组中删去,防止后面的重复计数。
第二天、第三天的搜索情况同理。
AC代码:
#include<iostream>
#include<cstdio>
#include<map>
#include<algorithm>
#include<vector>
using namespace std;
const int maxn=2e5+; int n,m,d;
int a[maxn];
map<int ,int >ans;//if ans[x]=y : 时间点为x的咖啡在第y天喝
map<int ,int >b;//first存储咖啡时间点,second随意 void Solve()
{
for(int i=;i <= n;++i)
b[a[i]]=i; int day=;
int tot=;
map<int ,int >::iterator it; for(it=b.begin();it != b.end();++it)
{
//当前有tot个时间点已求出解
if(tot >= n)
break; int x=it->first;
if(!ans.count(x))//如果在之前的day中没有筛去x时间点,那么x时间点在第day+1天喝
{
ans[x]=++day;
tot++;
}
//查找出第day天可以喝的咖啡时间点
map<int ,int >::iterator t=b.lower_bound(x+d+);
while(t->first >= x+d+)
{
ans[t->first]=ans[x];
x=t->first;
b.erase(t); t=b.lower_bound(x+d+);
tot++;
}
} printf("%d\n",day);
for(int i=;i <= n;++i)
printf("%d ",ans[a[i]]);
}
int main()
{
scanf("%d%d%d",&n,&m,&d);
for(int i=;i <= n;++i)
scanf("%d",a+i);
Solve(); return ;
}