算法学习 - 最小栈的实现O(1)时间

1. //
2. //  main.cpp
3. //  MinStack2
4. //
5. //  Created by Alps on 14/12/3.
6. //  Copyright (c) 2014年 chen. All rights reserved.
7. //
8. #include <iostream>
9. #include <vector>
10. using namespace std;
11. class MinStack {
12. public:
13. vector<int> stack;
14. int min;
15. void push(int x) {
16. if (stack.empty()) {
17. stack.push_back(0);
18. min = x;
19. }else{
20. stack.push_back(x-min);
21. if (x < min) {
22. min = x;
23. }
24. }
25. }
26. void pop() {
27. if (stack.empty()) {
28. return;
29. }else{
30. if (stack.back() < 0) {
31. min = min - stack.back();
32. }
33. stack.pop_back();
34. }
35. }
36. int top() {
37. if (stack.empty()) {
38. return NULL;
39. }
40. if (stack.back() > 0) {
41. return stack.back()+min;
42. }else{
43. return min;
44. }
45. }
46. int getMin() {
47. if (stack.empty()) {
48. return NULL;
49. }
50. return min;
51. }
52. };
53. int main(int argc, const char * argv[]) {
54. // insert code here...
55. std::cout << "Hello, World!\n";
56. return 0;
57. }

在边缘测试的时候，我的编译器没出错，但是OJ上报WA了。

1. //
2. //  main.cpp
3. //  MinStack4_leetcode
4. //
5. //  Created by Alps on 14/12/3.
6. //  Copyright (c) 2014年 chen. All rights reserved.
7. //
8. #include <iostream>
9. #include <stack>
10. using namespace std;
11. class MinStack{
12. public:
13. stack<long> s;
14. long min;
15. void push(int x){
16. if (s.empty()) {
17. s.push(0);
18. min = x;
19. }else{
20. s.push(x-min);
21. if (x < min) {
22. min = x;
23. }
24. }
25. }
26. void pop(){
27. if (s.empty()) {
28. return;
29. }else{
30. if (s.top() < 0) {
31. min = min - s.top();
32. }
33. s.pop();
34. }
35. }
36. int top(){
37. if (s.empty()) {
38. return NULL;
39. }else{
40. if (s.top() > 0) {
41. return (int)(min+s.top());
42. }else{
43. return (int)min;
44. }
45. }
46. }
47. int getMin(){
48. if (s.empty()) {
49. return NULL;
50. }else{
51. return (int)min;
52. }
53. }
54. };
55. int main(int argc, const char * argv[]) {
56. int a = -2147483648;
57. MinStack M;
58. M.push(2147483646);
59. M.push(2147483646);
60. M.push(2147483647);
61. printf("%d\n",M.top());
62. M.pop();
63. printf("%d\n",M.getMin());
64. M.pop();
65. printf("%d\n",M.getMin());
66. M.pop();
67. M.push(2147483647);
68. printf("%d\n",M.top());
69. printf("%d\n",M.getMin());
70. M.push(a);
71. printf("%d\n",M.top());
72. printf("%d\n",M.getMin());
73. M.pop();
74. printf("%d\n",M.getMin());
75. return 0;
76. }

自己写的链表结构体。

1. //
2. //  main.cpp
3. //  MinStack3_leetcode
4. //
5. //  Created by Alps on 14/12/3.
6. //  Copyright (c) 2014年 chen. All rights reserved.
7. //
8. #include <iostream>
9. using namespace std;
10. class MinStack {
11. public:
12. struct StackNode{
13. int num;
14. StackNode *next;
15. };
16. typedef StackNode* stack;
17. stack s;
18. MinStack(){
19. s = (stack)malloc(sizeof(struct StackNode));
20. s->next = NULL;
21. }
22. int min;
23. void push(int x) {
24. if (s->next == NULL) {
25. stack node = (stack)malloc(sizeof(struct StackNode));
26. node->num = x;
27. node->next = s->next;
28. s->next = node;
29. min = x;
30. }else{
31. stack node = (stack)malloc(sizeof(struct StackNode));
32. node->num = x;
33. node->next = s->next;
34. s->next = node;
35. if (x < min) {
36. min = x;
37. }
38. }
39. }
40. void pop() {
41. if (s->next == NULL) {
42. return;
43. }else{
44. stack temp = s->next;
45. if (min == s->next->num && s->next->next != NULL) {
46. s->next = s->next->next;
47. free(temp);
48. min = s->next->num;
49. for (stack loop = s->next; loop != NULL; loop = loop->next) {
50. if (min > loop->num) {
51. min = loop->num;
52. }
53. }
54. }else{
55. s->next = s->next->next;
56. free(temp);
57. }
58. }
59. }
60. int top() {
61. if (s->next == NULL) {
62. return NULL;
63. }
64. return s->next->num;
65. }
66. int getMin() {
67. if (s->next == NULL) {
68. return NULL;
69. }
70. return min;
71. }
72. };
73. int main(int argc, const char * argv[]) {
74. MinStack MinS;
75. MinS.push(-1);
76. MinS.push(0);
77. MinS.push(2);
78. MinS.push(-2);
79. printf("%d\n",MinS.top());
80. MinS.pop();
81. MinS.pop();
82. MinS.pop();
83. printf("%d\n",MinS.getMin());
84. return 0;
85. }

最古老的版本。

1. //
2. //  main.cpp
3. //  MinStack_leetcode
4. //
5. //  Created by Alps on 14/12/2.
6. //  Copyright (c) 2014年 chen. All rights reserved.
7. //
8. #include <iostream>
9. #include "vector"
10. using namespace std;
11. class MinStack {
12. public:
13. struct StackNode{
14. int num;
15. int min;
16. };
17. vector<StackNode> stack;
18. void push(int x) {
19. if (stack.empty()) {
20. StackNode S = {x,x};
21. stack.push_back(S);
22. }else{
23. if (x < stack.back().min) {
24. StackNode S = {x,x};
25. stack.push_back(S);
26. }else{
27. StackNode S = {x,stack.back().min};
28. stack.push_back(S);
29. }
30. }
31. }
32. void pop() {
33. if (stack.empty()) {
34. }else{
35. stack.pop_back();
36. }
37. }
38. int top() {
39. if (stack.empty()) {
40. return NULL;
41. }
42. return stack.back().num;
43. }
44. int getMin() {
45. if (stack.empty()) {
46. return NULL;
47. }
48. return stack.back().min;
49. }
50. };
51. int main(int argc, const char * argv[]) {
52. MinStack minstack;
53. minstack.push(-1);
54. minstack.push(1);
55. printf("%d %d\n",minstack.top(), minstack.getMin());
56. return 0;
57. }