[ACM] 406 Prime Cuts

題目406
寫得有點冗長，而且在ACM居然沒過。
但在其他JUDGE過了。

找時間再來修正

1. #include <iostream>
2. #include <vector>
3. using namespace std;
4. const int MAX = 1000;
5. void creat(bool* prime);
6. void filter(bool*, int);
8. int main()
9. {
10.     bool prime[MAX + 1] = {};
11.     creat(prime);
12.     int n1,c;
14.     while(cin >> n1 >> c)
15.     {
    
16.         if ( n1 == 1 )
17.         {
18.             cout << n1 << ' ' << c << ": " << "1\n" ;
19.             continue;
20.         }
21.         vector<int> box;
22.         for (int i=1; i<=n1 ; i++)
23.         {
24.             if (prime[i])
25.                 box.push_back(i);
26.         }
27.         int len = box.size();
29.         if (len % 2 == 1)
30.         {
31.             if ( 2*c-1 > len)
32.             {
33.                 cout << n1 << ' ' << c << ": ";
34.                 for (int i=0; i<len; i++)
35.                     cout << box[i] << " ";
36.             }
37.             else
38.             {
39.             int mid = len/2, begin = mid-c+1, end = mid+c-1;
40.             cout << n1 << ' ' << c << ": ";
41.             for (int i=begin; i<=end; i++)
42.                 cout << box[i] << " ";
43.             }
44.         }
45.         else //len % 2 == 0
46.         {
47.             if ( 2*c > len)
48.                 {
49.                     cout << n1 << ' ' << c << ": ";
50.                     for (int i=0; i<len; i++)
51.                         cout << box[i] << " ";
52.                 }
53.             else
54.             {
55.                 int mid = len/2, begin = mid-c, end = mid+c-1;
56.                 cout << n1 << ' ' << c << ": ";
57.                 for (int i=begin; i<=end; i++)
58.                     cout << box[i] << " ";
59.             }
60.         }
61.         cout << endl;
62.     }
64. }
65. //創造質數表
66. void creat(bool* prime)
67. {
68.     prime[2] = prime[3] = prime[5] = 1;
69.     //把6n+1、6n+5全設為質數
70.     int i;
71.     for (i=1; 6*i+5 <= MAX; i++)
72.     {
73.         prime[6*i+5] = prime[6*i+1] = 1;
74.     }
75.     if (prime[6*i+1] <= MAX)
76.     {
77.         prime[6*i+1] = 1;
78.     }
79.     //然後過濾掉6n+1、6n+5的倍數皆不為質數
80.     int j;
81.     for (j=0; (6*j+5)*(6*j+5) <= MAX; j++)
82.     {
83.        filter(prime, 6*j+5);
84.        filter(prime, 6*j+1);
85.     }
86.     if (prime[6*j+1] <= MAX)
87.     {
88.         filter(prime, 6*j+1);
89.     }
90.     prime[1] = 1;
91. }
92. //過濾掉質數的倍數
93. void filter(bool* prime, int i)
94. {
95.     if (prime[i])
96.     {
97.         for (int j=2; i*j <= MAX; j++)
98.             prime[i*j] = 0;
99.     }
100. }