PAT-乙-1069 1069 微博转发抽奖 (20 分)
代码
#include <iostream>
#include <vector>
#include <map>
using namespace std;
int main() {
int m, n, s;
cin>>m>>n>>s;
string str;
getline(cin, str);
vector<string> v;
for(int i=0; i<m; i++){
getline(cin, str);
v.push_back(str);
}
map<string, int> map;
vector<string> ans;
for(int i=s-1; i<m; ){
if(map[v.at(i)]){
i++;
}
else{
map[v.at(i)] = 1;
ans.push_back(v.at(i));
i+=n;
}
}
if(ans.size()==0){
cout<<"Keep going..."<<endl;
}
else{
for(int i=0; i<ans.size(); i++){
cout<<ans.at(i)<<endl;
}
}
return 0;
}
注解
1、当cin与getline混用时,一定不要忘记读行末尾的空格。
如代码中的
cin>>m>>n>>s;
string str;
getline(cin, str);
先读入m,n,s,下面的getline其实是读入当行的剩余字符。
2、核心处理过程:首先从s-1开始,看当前人是否已经中奖,如果是,则取下一位。否则,取当前人为中奖。每隔N人抽中一个,所以i每次需要加n,而不是通常的i++。
for(int i=s-1; i<m; ){
if(map[v.at(i)]){
i++;
}
else{
map[v.at(i)] = 1;
ans.push_back(v.at(i));
i+=n;
}
}