使用队列求解杨辉三角的第K层的所有元素

Java代码  

1. package queue;
2. import java.util.concurrent.ConcurrentLinkedDeque;
3. /**
4. * Created by Lanxiaowei
5. * Craated on 2016/12/12 9:03
6. * 求解杨辉三角的第K层的所有元素
7. * 使用队列求解
8. */
9. public class YHTriangleWithQueue {
10. public static void main(String[] args) throws InterruptedException {
11. int k = 6;
12. int[][] nums = kTier(k);
13. for(int j=0; j < k; j++) {
14. System.out.print(nums[k % 2][j] + " ");
15. }
16. }
17. 下载
18. public static int[][] kTier(int k) throws InterruptedException {
19. int[][] nums = new int[2][k];
20. if(k <= 0) {
21. throw new IllegalArgumentException("Argument k MUST be > 0.");
22. }
23. ConcurrentLinkedDeque<Event> queue = new ConcurrentLinkedDeque<Event>();
24. Event head = new Event(1,1);
25. Event tail = new Event();
26. int level = 0;
27. int pos = 0;
28. queue.add(head);
29. while (!queue.isEmpty()) {
30. head = queue.getFirst();
31. level = head.getLevel();
32. pos = head.getPos();
33. if(pos == 1 || pos == level) {
34. nums[level % 2][pos - 1] = 1;
35. } else {
36. nums[level % 2][pos - 1] = nums[(level - 1) % 2][pos - 1] + nums[(level - 1) % 2][pos - 2];
37. }
38. if(level < k) {
39. Event tt = new Event(level + 1,pos);
40. queue.addLast(tt);
41. if(pos == level) {
42. Event t = new Event(tt.getLevel(),pos + 1);
43. queue.addLast(t);
44. }
45. }
46. queue.pop();
47. }
48. return nums;
49. }
50. 下载
51. public static class Event {
52. //表示第几层
53. private int level;
54. //第几位
55. private int pos;
56. public Event() {}
57. public Event(int level, int pos) {
58. this.level = level;
59. this.pos = pos;
60. }
61. public int getLevel() {
62. return level;
63. }
64. public void setLevel(int level) {
65. this.level = level;
66. }
67. public int getPos() {
68. return pos;
69. }
70. public void setPos(int pos) {
71. this.pos = pos;
72. }
73. }
74. }