链接:https://leetcode.cn/problems/number-of-students-unable-to-eat-lunch/solutions/1903212/mo-ni-zhu-shi-chao-ji-xiang-xi-by-xun-ge-gimg/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
题目
示例
思路
题意 -> 用队列中的元素匹配栈中元素,返回不能匹配的元素
每一轮匹配原则:
- 如果队列最前面的学生 喜欢 栈顶的三明治,那么会 拿走它 并离开队列。相当于栈顶元素出栈
- 否则,这名学生会 放弃这个三明治 并回到队列的尾部。
最后返回栈中剩余元素
可以使用模拟枚举栈和队列中的所以元素,逐一进行匹配,当队列中所有元素都不能匹配栈顶元素时结束。
优化在上述操作中需要对队列进行反复操作,但是可以发现在匹配过程中,队列中的所有元素都是可以移动的,并且任意一个元素的可以做对头,而题目只需要有多少个同学没有吃的,并不关心具体是哪个同学,那么我们可以统计所有学生喜欢0和1的个数,直接与栈进行匹配。
代码
int countStudents(int* students, int studentsSize, int* sandwiches, int sandwichesSize){
int student[2] = {0};
//统计学生喜欢 0 和 1的个数
for(int i = 0; i < studentsSize; i++)
{
student[students[i]]++;
}
//与栈顶进行匹配,一旦匹配不成功结束
for(int i = 0; i < sandwichesSize; i++)
{
if(student[sandwiches[i]]--);
else
return sandwichesSize-i;
}
return 0;
}
作者:小迅
链接:https://leetcode.cn/problems/number-of-students-unable-to-eat-lunch/solutions/1903212/mo-ni-zhu-shi-chao-ji-xiang-xi-by-xun-ge-gimg/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。