7-5 jmu-報數遊戲 （15 分)

報數遊戲是這樣的：有n我的圍成一圈，按順序從1到n編好號。從第一我的開始報數，報到m（m<n）的人退出圈子；下一我的從1開始報數，報到m的人退出圈子。如此下去，直到留下最後一我的。其中n是初始人數；m是遊戲規定的退出位次（保證爲小於n的正整數）。要求用隊列結構完成。輸出數字間以空格分隔，但結尾不能有多餘空格。

輸入樣例:

5 3

輸出樣例:

3 1 5 2 4

輸入樣例:

5 6

輸出樣例:

看到這個題，想到是用STL set來作
下面是最開始的代碼error!

#include <iostream> #include<set>
using namespace std; int main() { set<int> s; int m, n; cin >> n >> m; if (m >= n) { cout << "error!"; return 0; } for (int i = 0; i < n; i++) { s.insert(i + 1); } set<int>::iterator it; it = s.begin(); int flag = 0; while (s.size() != 0) { for (int i = 1; i < m; i++) { if (!(it != s.end())) { it = s.begin(); } it++; } if (!(it != s.end())) { it = s.begin(); } if (flag == 0) { cout << *it; flag = 1; } else { cout << " " << *it; } int c = *it; s.erase(c); int j = 1; it = s.find((c + 1) % n); if (!(it != s.end())) { it = s.begin(); } } }

測試結果：

 

可是看這一句  it = s.find((c + 1) % n);  假設（c+1）%n在set中沒有找到的話，返回給it的地址就是s.end()  可是假設c+1沒找到後面還有c+2的話就會出錯，因此我手寫了一個測試點，n=8，m=3。

按照題目意思推理得出： 3 6 1 5 2 8 4 7

可是按照我寫的代碼得出的結果是：3 6 1 5 7 8 2 4

因此改進版代碼爲：

#include <iostream> #include<set>
using namespace std; int main() { set<int> s; int m, n; cin >> n >> m; if (m >= n) { cout << "error!"; return 0; } for (int i = 0; i < n; i++) { s.insert(i+1); } set<int>::iterator it; it = s.begin(); int flag = 0; while (s.size() != 0) { for (int i = 1; i < m; i++) { if (!(it != s.end())) { it = s.begin(); } it++; } if (!(it != s.end())) { it = s.begin(); } if (flag == 0) { cout << *it; flag = 1; } else { cout <<" "<< *it; } int c = *it; s.erase(c); int j = 1; while (1) { it = s.find((c + j) % n); if (it != s.end()) break; else j++; if (j == n) { break; it = s.end(); } } if (!(it != s.end())) { it = s.begin(); } } }

得出結果：3 6 1 5 2 8 4 7

PTA提交結果：正確

同一個測試點兩種代碼給出的結果是不同的，而PTA上提交都是正確的，說明測試點沒有考慮到n遠大於m的狀況