题目链接:E. Pencils and Boxes
题意:
给出n个数字,分组,每组大小>=k,并且同一组内任意两个数字大小不能超过d
思路:
先从小到大排序。然后dp。起点肯定是第一个数字,然后判断能够和它组成第一个组的下标位置,将这些位置的dp标记为1,然后从(这些位置+1)的地方作为起点继续往下找终点,找到能作为终点的位置都标记为1,然后往复,一直到结束。如果dp[n]被标记成了1就说明能够分组成功。输出YES,否注就NO。
代码:
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
long long a[500005];
int dp[500005];
int main()
{
int n,k,i;
long long d;
cin>>n>>k>>d;
for(i=1;i<=n;i++)
cin>>a[i];
sort(a+1,a+n+1);
int ind=1;
memset(dp,0,sizeof(dp));
dp[0]=1;
for(i=0;i<=n;++i)
{
if(dp[i])
{
//搜索过的就去掉 避免重复 不去掉一些特例会超时
ind=max(ind,i+k);
//这边有个坑 如果数组开的不够大,一定要先判断下标 否注会越界。找了很久才发现
while(ind<=n&&a[ind]-a[i+1]<=d)
{
dp[ind]=1;
ind++;
}
}
}
if(dp[n])cout<<"YES";
else cout<<"NO";
return 0;
}