数糖纸（C++）

最新推荐文章于 2024-12-07 22:36:46 发布

原创最新推荐文章于 2024-12-07 22:36:46 发布 · 845 阅读

2 ·

CC 4.0 BY-SA版权

牛客网专栏收录该内容

16 篇文章

订阅专栏

本文介绍了一种算法问题，围绕怪人PM6收集不同颜色且不重复的糖纸，通过队列数据结构优化求解最大收集数量。关键在于使用队列跟踪不重复的颜色，并更新最长连续无重复颜色的序列长度。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述

怪人PM6喜欢数糖纸，不同的糖纸有不同的颜色，一共有 N 张糖纸，第 i 张糖纸颜色为 Ci ，它们的位置都是固定的。PM6喜欢五彩缤纷的糖纸，所以他不希望有重复的颜色。他有一次机会，可以收集任意一段连续区间内的糖纸。求出PM6最多能收集多少张糖纸。

输入描述:

第一行一个正整数 N ，表示共有 N 张糖纸。
第二行共有 N 个正整数，第 i 个正整数表示第 i 张糖纸的颜色 Ci
对于20%的数据：1<=N<=100
对于40%的数据：1<=N<=1000
对于100%的数据：1<=N<=1e6，0<=Ci<=1e9

输出描述:

一个整数表示PM6最多能收集多少张糖纸。

示例1

输入

5
1 2 2 3 4

输出

说明

PM6可以收集第3到第5张的糖纸，共有三张。

注：对题面有删改，但意思没变。

不难发现，这个题用队列来做，不过要注意的是这个题的空间。
按样例来看：

1
//将1放入队列
1 2
//将2放入队列
1 2 2
//将2放入队列，发现之前有个2了
2 2
//将1出队，发现队列中还有相同的数
2
//将2出队
2 3
//将3入队
2 3 4
//将4入队

最后，我们发现符合条件的最长队列长度是3，故答案为3。
代码：

#include<bits/stdc++.h>
using namespace std;
int n,a[1000001],ans=0,len=0;
queue<int> q;
bool ok[1000000001];
int main() {
	scanf("%d",&n);
	for(int i=1; i<=n; i++) {
		scanf("%d",&a[i]);
	}
	for(int i=1; i<=n; i++) {
		q.push(a[i]);
		if(!ok[a[i]]) {
			ok[a[i]]=1;
			len++;
		} else {
			while(q.front()!=a[i]) {
				ok[q.front()]=0;
				q.pop();
				len--;
			}
			q.pop();
		}
		ans=max(ans,len);
	}
	printf("%d\n",ans);

	return 0;
}