【C++二分查找】825. 适龄的朋友

最新推荐文章于 2024-10-17 21:05:59 发布

闻缺陷则喜何志丹

最新推荐文章于 2024-10-17 21:05:59 发布

阅读量903

点赞数 21

文章标签： c++ 算法 leetcode 二分查找朋友请求数量

本文链接：https://blog.csdn.net/he_zhidan/article/details/140890358

版权

工作级难度算法题解专栏收录该内容

137 篇文章 0 订阅

订阅专栏

本文涉及的基础知识点

C++二分查找

LeetCode825. 适龄的朋友

在社交媒体网站上有 n 个用户。给你一个整数数组 ages ，其中 ages[i] 是第 i 个用户的年龄。
如果下述任意一个条件为真，那么用户 x 将不会向用户 y（x != y）发送好友请求：
ages[y] <= 0.5 * ages[x] + 7
ages[y] > ages[x]
ages[y] > 100 && ages[x] < 100
否则，x 将会向 y 发送一条好友请求。
注意，如果 x 向 y 发送一条好友请求，y 不必也向 x 发送一条好友请求。另外，用户不会向自己发送好友请求。
返回在该社交媒体网站上产生的好友请求总数。
示例 1：
输入：ages = [16,16]
输出：2
解释：2 人互发好友请求。
示例 2：
输入：ages = [16,17,18]
输出：2
解释：产生的好友请求为 17 -> 16 ，18 -> 17 。
示例 3：
输入：ages = [20,30,100,110,120]
输出：3
解释：产生的好友请求为 110 -> 100 ，120 -> 110 ，120 -> 100 。

提示：
n == ages.length
1 <= n <= 2 * 10⁴
1 <= ages[i] <= 120

二分查找

如果条件三成立，则条件二一定成立，故条件三是多余的。
同时满足以下两个条件则发送：
条件四：ages[y] > 0.5*ages[x]+7
条件五：ages[y] <= ages[x]
条件四简化后：2 $\times$ ages[y] > ages[x]+14 即 ages[x] < 2 $\times$ ages[y]-14
联合后，条件六：ages[y] <= arges[x] < 2 $\times$ ages[y]-14
向对ages排序，枚举ages[y]，统计符合条件的ages[x]。由于ages[y]一定符合，故要扣掉自身。最后的返回值减去ages.size()。
一个错误：
如果n >=2n-14，需要忽略。否则：a,数量可能是负数。b，自身不符合。不忽略的才需要扣除本身。

代码

核心代码

class Solution {
		public:
			int numFriendRequests(vector<int>& ages) {
				sort(ages.begin(), ages.end());
				int ret = 0;
				for (const auto& n : ages) {
					if (n >= 2* n - 14) { continue; }
					auto it1 = lower_bound(ages.begin(), ages.end(), n);
					auto it2 = lower_bound(ages.begin(), ages.end(), 2*n-14);
					ret += it2 - it1;		
					ret--;	
				}
				return ret ;
			}
			
		};

单元测试用例

TEST_CLASS(LeetCode825)
	{
		//LeetCode825. 适龄的朋友
		class Solution {
		public:
			int numFriendRequests(vector<int>& ages) {
				sort(ages.begin(), ages.end());
				int ret = 0;
				for (const auto& n : ages) {
					if (n >= 2* n - 14) { continue; }
					auto it1 = lower_bound(ages.begin(), ages.end(), n);
					auto it2 = lower_bound(ages.begin(), ages.end(), 2*n-14);
					ret += it2 - it1;		
					ret--;	
				}
				return ret ;
			}
			int Test(vector<int>& ages) {
				int ret = 0;
				sort(ages.begin(), ages.end());
				for (int y = 0; y < ages.size(); y++) {
					Logger::WriteMessage(("\r\n" + std::to_string(ages[y])+"\r\n").c_str());
					for (int x = 0; x < ages.size(); x++) {
						if (y == x) { continue; }
						if (ages[y] <= 0.5 * ages[x] + 7) { continue; }
						if (ages[y] > ages[x]) { continue; }
						ret++;
						Logger::WriteMessage((" " + std::to_string(ages[x])).c_str());
					}
				}
				return ret;
			}
		};
	public:
		vector<int> ages;
		TEST_METHOD(TestMethod1)
		{
			ages = { 5 };
			auto res = Solution().numFriendRequests(ages);
			AssertEx(0, res);
		}
		TEST_METHOD(TestMethod2)
		{
			ages = { 20 };
			auto res = Solution().numFriendRequests(ages);
			AssertEx(0, res);
		}
		TEST_METHOD(TestMethod3)
		{
			ages = { 11,12 };
			auto res1 = Solution().Test(ages);
			auto res = Solution().numFriendRequests(ages);
			AssertEx(res1, res);
		}
		TEST_METHOD(TestMethod11)
		{
			ages = { 16, 16 };
			auto res = Solution().numFriendRequests(ages);
			AssertEx(2, res);
		}
		TEST_METHOD(TestMethod12)
		{
			ages = { 16, 17,18 };
			auto res = Solution().numFriendRequests(ages);
			AssertEx(2, res);
		}
		TEST_METHOD(TestMethod13)
		{
			ages = { 20,30,100,110,120 };
			auto res = Solution().numFriendRequests(ages);
			AssertEx(3, res);
		}
		TEST_METHOD(TestMethod14)
		{
			ages = { 108,115,5,24,82 };
			auto res = Solution().numFriendRequests(ages);
			AssertEx(3, res);
		}
		TEST_METHOD(TestMethod15)
		{
			ages = { 101,98,80,20,1,97,3,77,114,109 };			
			auto res1 = Solution().Test(ages);
			auto res = Solution().numFriendRequests(ages);
			AssertEx(res1, res);
		}
	};

扩展阅读

我想对大家说的话
工作中遇到的问题，可以按类别查阅鄙人的算法文章，请点击《算法与数据汇总》。
学习算法：按章节学习《喜缺全书算法册》，大量的题目和测试用例，打包下载。重视操作
有效学习：明确的目标及时的反馈拉伸区（难度合适）专注
闻缺陷则喜(喜缺)是一个美好的愿望，早发现问题，早修改问题，给老板节约钱。
子墨子言之：事无终始，无务多业。也就是我们常说的专业的人做专业的事。
如果程序是一条龙，那算法就是他的是睛
失败+反思=成功成功+反思=成功