编程练习（小s的第x个苹果）

最新推荐文章于 2023-09-28 12:55:54 发布

x缩

最新推荐文章于 2023-09-28 12:55:54 发布

阅读量195

点赞数

分类专栏：编程练习（各种题目）

本文链接：https://blog.youkuaiyun.com/qq_43230068/article/details/89311346

版权

编程练习（各种题目）专栏收录该内容

10 篇文章

订阅专栏

该博客探讨了一道编程题目，涉及n堆苹果，寻找第x个苹果所在的具体堆。博主首先介绍了初始的遍历解法，但发现效率低下，仅能通过部分测试。随后，博主采用了二分查找法优化算法，成功通过了所有测试案例。

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

题目：有从左到右的n堆苹果。小s知道每堆苹果的个数，那么请问第x个苹果是从左往右数第几堆中的呢？

要求：
输入
第一行： n（代表有n堆苹果，n>=1）
第二行： a1，a2。。。。an。（ai代表第i堆里有ai个苹果）
第三行： m（询问m次）
第四行： q1，q2。。。。qm。（qi表示第i次询问第qi个苹果在第几堆）

思路：
创建一个vector sum，sum[i]的表示前i堆苹果的总数，将q与sum[i]比较。
一开始是用遍历的方法比较q与sum[i](i从0到n），只能通过30%的测试，计算量太大。
后来使用二分查找法比较q与sum，通过了所有测试。

通过测试的代码：

#include<iostream>
#include<vector>
using namespace std;
int main()
{
	unsigned int n;
	unsigned int m;
	vector<int> a;
	a.push_back(0);
	vector<int> q;
	int sum = 0;
	int tmp;
	cin >> n;
	getchar();
	while (cin >> tmp)
	{
		sum += tmp;
		a.push_back(sum);
		if (getchar() == '\n')
			break;
	}
	cin >> m;
	getchar();
	while (cin >> tmp)
	{
		q.push_back(tmp);
		if (getchar() == '\n')
			break;
	}
	for (unsigned int i = 0; i < m; i++)
	{
		if (q[i] > a[a.size()-1])
		{
			cout << " " << endl;
			continue;
		}
		unsigned int low = 0;
		unsigned int high = a.size()-1;
		unsigned int mid = (low + high) / 2;
		while (!(q[i]<a[mid] && q[i]>a[mid - 1]))
		{
			if (q[i] == a[mid])
			{
				break;
			}
			else if(q[i]>a[mid])
			{
				low = mid + 1;
				mid = (low + high) / 2;
			}
			else if(q[i]<a[mid])
			{
				high = mid-1;
				mid = (low + high) / 2;
			}
		}
		cout << mid << endl;
	}
	return 0;
}