E. Pencils and Boxes codeforces(思维+dp)

题目链接: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;
}