CCF - 201709-2 - 公共钥匙盒

 1 #include <iostream>
 2 #include <algorithm>
 3 using namespace std;
 4 
 5 int main()
 6 {
 7     int N, K;                        // N 把钥匙， K 个老师
 8     int w[1001], s[1001], c[1001];   // w 钥匙编号，s 初始时间，c 结束时间
 9     int n[1001] = {};                // 钥匙架 空为0
10     int time = 1;                    // 时间轴
11     int end_time = 0;
12     int i, j;
13 
14     /* 初始化 */
15     cin>>N>>K;
16     for(i = 0; i < K; i ++)
17     {
18         cin>>w[i]>>s[i]>>c[i];
19         c[i] += s[i];
20         end_time = end_time > c[i] ? end_time : c[i];
21     }
22     for(i = 1; i <= N; i ++)
23     {
24         n[i] = i;
25     }
26 
27     /* 过程 */
28     while(time <= end_time)
29     {
30         int back[1001] = {};    // 还钥匙序列
31         int num = 0;
32         // 还钥匙
33         for(i = 0; i < K; i ++)
34         {
35             if(c[i] == time)
36             {
37                 back[num] = w[i];
38                 num ++;
39             }
40         }
41         if(num != 0)
42         {
43             sort(back, back + num);
44             j = 0;
45             for(i = 1; i <= N; i ++)
46             {
47                 if(n[i])
48                     continue;
49                 n[i] = back[j];
50                 j ++;
51                 if(j == num)
52                     break;
53             }
54         }
55         // 取钥匙
56         for(i = 0; i < K; i ++)
57         {
58             if(s[i] == time)
59             {
60                 for(j = 1; j <= N; j ++)
61                 {
62                     if(n[j] == w[i])
63                         n[j] = 0;
64                 }
65             }
66         }
67         time ++;
68     }
69 
70     /* 结果 */
71     for(i = 1; i <= N; i ++)
72     {
73         cout<<n[i]<<" ";
74     }
75 
76     return 0;
77 }