B. Coloring【Polynomial Round 2022 (Div. 1 + Div. 2, Rated, Prizes!)】

题目

B. Coloring

做法

首先，我们可以将$n$个格子分成$\lceil \frac{n}{k}\rceil$段 ，除了最后一段外，所有段的长度都为$k$。然后，根据题意，要求在每段中，每个格子的颜色互不相同。很容易发现任何$a_i$应小于或等于$\lceil \frac{n}{k}\rceil$。
例如：
$n$ $m$ $k$
$13$ $6$ $3$
$A$ $B$ $C$ $D$ $E$ $F$
$1$ $1$ $1$ $1$ $4$ $5$
将$13$个格子分成$\lceil \frac{13}{3}\rceil =5$段，除了最后一段外，所有段的长度都为$3$。然后在每段中，每个格子的颜色互不相同：
$A$ $E$ $F$
$B$ $E$ $F$
$C$ $E$ $F$
$D$ $E$ $F$
$F$
很容易发现$a_i$应小于或等于$5$，这样每一段最多可以放一种颜色，最多放5种。
然后我们可以计算等于$\lceil \frac{n}{k}\rceil$的$a_i$的数量。此数字必须小于或等于$n$ $mod$ $k$，即最后一段的长度。
原因是，等于$\lceil \frac{n}{k}\rceil$的$a_i$一定要放在最后一段才能放得下，那么这样的$a_i$的数量不能大于最后一段的长度，这样的话，所有这样的$a_i$才能放得下。
满足上述条件的所有$a$均有效。我们可以使用以下方法构造着色：
首先，我们挑选所有满足$a_i=\lceil \frac{n}{k}\rceil$的颜色$i$，然后我们使用颜色$i$为每个片段中的第$j$个单元格着色。
然后我们挑选所有满足$a_i<\lceil \frac{n}{k}\rceil -1$的颜色$i$，并使用这些颜色以循环顺序对其余单元格着色（即：将颜色染上第一段的第$j$个单元格，第二段的第$j$个单元格，…，第$\lceil \frac{n}{k}\rceil$段的第$j$个单元格，然后让$j+1$。当一种颜色用完时，我们开始使用下一种颜色）
最后，我们挑选所有满足$a_i=\lceil \frac{n}{k}\rceil -1$的颜色，并用循环顺序对它们进行着色。
此方法将始终给出有效的构造。
例如：
$n$ $m$ $k$
$13$ $6$ $3$
$A$ $B$ $C$ $D$ $E$ $F$
$1$ $1$ $1$ $1$ $4$ $5$
$A$ $E$ $F$
$B$ $E$ $F$
$C$ $E$ $F$
$D$ $E$ $F$
$F$

代码

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll maxn = 1e5+5;
ll t,n,m,x,y,k;
ll a[maxn];
int main()
{
	cin>>t;
	while(t--){
		cin>>n>>m>>k;
		ll maxx=-1e18;
		for(int i=1;i<=m;i++)cin>>a[i];
		bool flag=true;
		for(int i=1;i<=m;i++){
			if(a[i]>ceil(n*1.0/k))flag=false;
		}
		ll cnt=0;
		for(int i=1;i<=m;i++){
			if(a[i]==ceil(n*1.0/k))cnt++;
		}
		if(flag==true&&cnt<=(n%k))puts("YES");
		else if(flag==true&&(n%k==0)&&cnt<=k)puts("YES");
		else puts("NO");
	}
	return 0;
}