Mishka received a gift of multicolored pencils for his birthday! Unfortunately he lives in a monochrome world, where everything is of the same color and only saturation differs. This pack can be represented as a sequence a1, a2, ..., an of n integer numbers — saturation of the color of each pencil. Now Mishka wants to put all the mess in the pack in order. He has an infinite number of empty boxes to do this. He would like to fill some boxes in such a way that:
- Each pencil belongs to exactly one box;
- Each non-empty box has at least k pencils in it;
- If pencils i and j belong to the same box, then |ai - aj| ≤ d, where |x| means absolute value of x. Note that the opposite is optional, there can be pencils i and j such that |ai - aj| ≤ d and they belong to different boxes.
Help Mishka to determine if it's possible to distribute all the pencils into boxes. Print "YES" if there exists such a distribution. Otherwise print "NO".
The first line contains three integer numbers n, k and d (1 ≤ k ≤ n ≤ 5·105, 0 ≤ d ≤ 109) — the number of pencils, minimal size of any non-empty box and maximal difference in saturation between any pair of pencils in the same box, respectively.
The second line contains n integer numbers a1, a2, ..., an (1 ≤ ai ≤ 109) — saturation of color of each pencil.
Print "YES" if it's possible to distribute all the pencils into boxes and satisfy all the conditions. Otherwise print "NO".
6 3 10 7 2 7 7 4 2
YES
6 2 3 4 5 3 13 4 10
YES
3 2 5 10 16 22
NO
In the first example it is possible to distribute pencils into 2 boxes with 3 pencils in each with any distribution. And you also can put all the pencils into the same box, difference of any pair in it won't exceed 10.
In the second example you can split pencils of saturations [4, 5, 3, 4] into 2 boxes of size 2 and put the remaining ones into another box.
把n个铅笔分配到任意多个盒子里,每个盒子至少有k个铅笔,且其中的铅笔满足|ai-aj|<=d. 问有没有合法的分配方案。
答案一定是排序后,分隔成的一些连续的区间。
用dp[i]=1表示前i个铅笔可以被合理放置,且第i个铅笔可以作为其中最后一个区间的右界。则dp[i+1]=1当且仅当存在0<=j<=i-k+1,使得a[i]-a[j]<=d且dp[j]=1.初始时dp[0]=1.
用two-pointer维护对于当前 i , a[i]-a[j]<=d且0<=j<=i-k+1的 j 的范围,用cnt表示这些合法的 j 当中,有多少满足dp[j]=1. 这样,dp的过程被优化到了O(n)。
#include <cstdio>
#include <iostream>
#include <string.h>
#include <string>
#include <map>
#include <queue>
#include <deque>
#include <vector>
#include <set>
#include <algorithm>
#include <math.h>
#include <cmath>
#include <stack>
#include <iomanip>
#define mem0(a) memset(a,0,sizeof(a))
#define meminf(a) memset(a,0x3f,sizeof(a))
using namespace std;
typedef long long ll;
typedef long double ld;
typedef double db;
const int maxn=500005,inf=0x3f3f3f3f;
const ll llinf=0x3f3f3f3f3f3f3f3f;
const ld pi=acos(-1.0L);
int a[maxn],dp[maxn];
int main() {
int n,k,d,i,j;
scanf("%d%d%d",&n,&k,&d);
for (i=1;i<=n;i++) {
scanf("%d",&a[i]);
}
sort(a+1,a+n+1);
dp[0]=1;
int l=0,cnt=0;
for (i=k;i<=n;i++) {
if (dp[i-k]) cnt++;
while (a[i]-a[l+1]>d&&i-l+1>k) {
if (dp[l]) cnt--;
l++;
}
if (cnt&&a[i]-a[l+1]<=d&&i-l+1>k) dp[i]=1;
}
if (dp[n]) printf("YES"); else printf("NO");
return 0;
}