洛谷题目 P1215

zltzlt

洛谷题目 P1215

原题




我的代码

1. #include <bits/stdc++.h>
   
2. using namespace std;
   
3. 
   
4. int a, b, c, visited[20][20][20];
   
5. set<int> res;
   
6. set<int>::iterator it;
   
7. 
   
8. void dfs(int cura, int curb, int curc)
   
9. {
   
10.     // 如果当前状态重复了就直接返回
    
11.     if (visited[cura][curb][curc])
    
12.         return;
    
13.     // 如果当前 a 桶为空就将 c 桶的状态保存
    
14.     if (!cura)
    
15.         res.insert(curc);
    
16.     visited[cura][curb][curc] = 1;
    
17.     if (cura)
    
18.     {
    
19.         // A 桶往 B 桶倒
    
20.         dfs(((b - curb) >= cura) ? 0 : cura - (b - curb),
    
21.             ((b - curb) < cura) ? b : (curb + cura), curc);
    
22.         // A 桶往 C 桶倒
    
23.         dfs(((c - curc) >= cura) ? 0 : cura - (c - curc),
    
24.             curb, ((c - curc) < cura) ? c : (curc + cura));
    
25.     }
    
26.     if (curb)
    
27.     {
    
28.         // B 桶往 A 桶倒
    
29.         dfs(((a - cura) < curb) ? a : (curb + cura),
    
30.             ((a - cura) >= curb) ? 0 : curb - (a - cura), curc);
    
31.         // B 桶往 C 桶倒
    
32.         dfs(cura, ((c - curc) >= curb) ? 0 : curb - (c - curc),
    
33.             ((c - curc) < curb) ? c : (curb + curc));
    
34.     }
    
35.     if (curc)
    
36.     {
    
37.         // C 桶往 A 桶倒
    
38.         dfs(((a - cura) < curc) ? a : (curc + cura), curb,
    
39.             ((a - cura) >= curc) ? 0 : curc - (a - cura));
    
40.         // C 桶往 B 桶倒
    
41.         dfs(cura, ((b - curb) < curc) ? b : (curc + curb),
    
42.             ((b - curb) >= curc) ? 0 : curc - (b - curb));
    
43.     }
    
44. }
    
45. 
    
46. int main()
    
47. {
    
48.     cin >> a >> b >> c;
    
49.     dfs(0, 0, c);
    
50.     for (it = res.begin(); it != res.end(); it++)
    
51.         cout << *it << ' ';
    
52.     return 0;
    
53. }

复制代码