题目如下
小明 PAT 考了满分,高兴之余决定发起微博转发抽奖活动,从转发的网友中按顺序每隔 N 个人就发出一个红包。请你编写程序帮助他确定中奖名单。
输入格式:
输入第一行给出三个正整数 M(≤ 1000)、N 和 S,分别是转发的总量、小明决定的中奖间隔、以及第一位中奖者的序号(编号从 1 开始)。随后 M 行,顺序给出转发微博的网友的昵称(不超过 20 个字符、不包含空格回车的非空字符串)。
注意:可能有人转发多次,但不能中奖多次。所以如果处于当前中奖位置的网友已经中过奖,则跳过他顺次取下一位。
输出格式:
按照输入的顺序输出中奖名单,每个昵称占一行。如果没有人中奖,则输出 Keep going…。
输入样例 1:
9 3 2
Imgonnawin!
PickMe
PickMeMeMeee
LookHere
Imgonnawin!
TryAgainAgain
TryAgainAgain
Imgonnawin!
TryAgainAgain
输出样例 1:
PickMe
Imgonnawin!
TryAgainAgain
输入样例 2:
2 3 5
Imgonnawin!
PickMe
输出样例 2:
Keep going…
解题思路
STL是个好东西,set能够直接去重,所以这题利用set应该是个非常明智的选择。先将数据存放在vector中,然后利用循环每次取vector中的数据,设置初始值为S-1,然后将该数据存放在set中,如果set的长度发生改变表示该值并没有在set中出现,因此直接输出该值,否则的话继续从该值往下遍历到往set中存放时能够使得set发生长度变化为止(表示该值没有重复出现)。最后再记录该值,并且再利用该值+N得到下一轮迭代的序号。
源代码
#include<iostream>
#include<string>
#include<set>
#include<vector>
using namespace std;
int main()
{
int M,N,S;
vector<string> vec;
set<string> st; // 用于判断是否重复
cin>>M>>N>>S;
for(int i=0;i<M;i++)
{
string str;
cin>>str;
vec.push_back(str); // vector存放所有转发的ID
}
for(int i=S-1;i<vec.size();)
{
int size=st.size(); // 记录初始的set长度
st.insert(vec[i]); // 将中奖者插入set中
if(st.size()!=size) // 当set的长度发生变化的时候输出,表示该中奖者获奖
{
cout<<vec[i]<<endl;
i+=N; // 进行下一轮:序号=当前序号+N
}
else // 如果长度不发生变化,说明该中奖者重复拿奖,应该跳过
{
for(int j=i+1;j<vec.size();j++) // 从该重复中奖者之后继续进行循环,查找出下一个中奖者
{
st.insert(vec[j]);
if(st.size()!=size) // 再次判断set的大小是否改变
{
cout<<vec[j]<<endl;
i=j+N; // 中奖,进行下一轮,序号=当前序号+N
break; // 跳出内循环,进行下一轮循环
}
}
}
}
if(st.size()==0) // 如果set的长度为0表示没有中奖者
{
cout<<"Keep going...";
}
}