【每日一题】—— C. Tiles Comeback （Codeforces Round 888 (Div. 3)）

news2024/11/24 0:45:57

🌏博客主页：PH_modest的博客主页
🚩当前专栏：每日一题
💌其他专栏：
🔴 每日反刍
🟡 C++跬步积累
🟢 C语言跬步积累
🌈座右铭：广积粮，缓称王！

一.题目描述

题目大意

地上有n块k种不同颜色的瓷砖，你现在站在第一块瓷砖上，你可以往右跳任意数量的瓷砖，然后你会得到一条长度为p的路径，路径的长度就是你跳跃到的瓷砖的数量。(即题目中的c1，c3，c4，为一条路径，长度为3）
问你是否有可能得到一条长度为p的路径，使得：它以索引为n的瓷砖结束；p被k整除，路径被划分为长度恰好为k的块；每个区块中的瓷砖具有相同的颜色，相邻区块中的颜色不一定不同。

注意：相同颜色的瓷砖不是一定要在同一块里面。~~比赛的时候没注意到这一点，WA了好几发~~

题目链接

C. Tiles Comeback （Codeforces Round 888 (Div. 3)）

二.思路分析

只需要看第一块和最后一块瓷砖是否满足条件即可

三.代码展示

#include<iostream>
#include<algorithm>
#include<vector>
#include<cmath>
#include<map>
#define int long long

using namespace std;
int s[200020];
void solve()
{
	int n,k;
	cin>>n>>k;
	int nums=0;
	for(int i=0;i<n;i++)
	{
		cin>>s[i];
		if(s[i]==s[0])//记录第一个砖块颜色有多少个
		{
			nums++;
		}
	}
	//特判
	if(k==1&&n==1)//只有一种颜色并且只能跳一个
	{
		cout<<"YES"<<"\n";
		return;
	}
	if(nums<k)//如果第一个砖块颜色少于k个，那么肯定不满足条件
	{
		cout<<"NO"<<"\n";
		return;
	}
	int l=s[0];//记录第一个砖块的颜色
	int r=s[n-1];//记录最后一个砖块的颜色

	if(l==r&&nums>=k)//如果最后砖块和第一个砖块颜色相同，那么只要该砖块颜色大于k就肯定满足条件
	{
		cout<<"YES"<<"\n";
		return;
	}
	if(l==r&&nums<k)//这边好像和前面的特判重复了
	{
		cout<<"NO"<<"\n";
		return;
	}
	
	int numsl=0;//记录第一个砖块颜色的数量
	int numsr=0;//记录最后一个砖块颜色的数量
	int pl,pr;//记录位置
	for(int i=0;i<n;i++)
	{
		if(s[i]==l)
		{
			numsl++;
		}
		if(numsl==k)
		{
			pl=i;
			break;
		}
	}
	for(int i=n-1;i>=0;i--)
	{
		if(s[i]==r)
		{
			numsr++;
		}
		if(numsr==k)
		{
			pr=i;
			break;
		}
	}
	if(numsr!=k)//如果最后一个砖块颜色小于k肯定不满足条件
	{
		cout<<"NO"<<"\n";
		return;
	}
	if(pl<pr)//第一个砖块的第k个必须要在最后一个砖块的左边
	{
		cout<<"YES"<<"\n";
	}
	else
	{
		cout<<"NO"<<"\n";
	}

}

signed main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);

	int t;
	cin>>t;
	while(t--)
	{
		solve();
	}

	return 0;
}