1 | #include "stream.h" |
---|
2 | #include <unordered_map> |
---|
3 | #include <map> |
---|
4 | #include <cstdlib> |
---|
5 | #include <sys/time.h> |
---|
6 | |
---|
7 | long timeval_diff(struct timeval &tv1, struct timeval &tv2) |
---|
8 | { |
---|
9 | long ret = tv1.tv_usec - tv2.tv_usec; |
---|
10 | if(ret<0) ret += 1000000; |
---|
11 | return ret; |
---|
12 | } |
---|
13 | |
---|
14 | |
---|
15 | long change(int n, int max=50) { |
---|
16 | if(n<0) |
---|
17 | return 0; |
---|
18 | |
---|
19 | if(n==0) |
---|
20 | return 1; |
---|
21 | |
---|
22 | long ret = 0; |
---|
23 | if(n>=1 && max>=1) ret+=change(n-1, 1); |
---|
24 | if(n>=2 && max>=2) ret+=change(n-2, 2); |
---|
25 | if(n>=5 && max>=5) ret+=change(n-5, 5); |
---|
26 | if(n>=10 && max>=10) ret+=change(n-10, 10); |
---|
27 | if(n>=20 && max>=20) ret+=change(n-20, 20); |
---|
28 | if(n>=50 && max>=50) ret+=change(n-50, 50); |
---|
29 | return ret; |
---|
30 | } |
---|
31 | |
---|
32 | long changef(int n, int max=4) { |
---|
33 | if(n<0) |
---|
34 | return 0; |
---|
35 | |
---|
36 | long ret = 1; |
---|
37 | switch(max) { |
---|
38 | case 4: ret += changef(n-50, 4); |
---|
39 | case 3: ret += changef(n-20, 3); |
---|
40 | case 2: ret += changef(n-10, 2); |
---|
41 | case 1: ret += changef(n-5, 1); |
---|
42 | case 0: ret += changef(n-2, 0); |
---|
43 | } |
---|
44 | return ret; |
---|
45 | } |
---|
46 | |
---|
47 | long changedyn0(int n, int max=4) { |
---|
48 | if(n<0) |
---|
49 | return 0; |
---|
50 | |
---|
51 | if(max<0) |
---|
52 | return 1; |
---|
53 | |
---|
54 | static std::map<int,long> cache[5]; |
---|
55 | std::map<int,long>::iterator it = cache[max].find(n); |
---|
56 | if(it != cache[max].end()) |
---|
57 | return cache[max][n]; |
---|
58 | |
---|
59 | long ret = 1; |
---|
60 | switch(max) { |
---|
61 | case 4: ret += changedyn0(n-50, 4); |
---|
62 | case 3: ret += changedyn0(n-20, 3); |
---|
63 | case 2: ret += changedyn0(n-10, 2); |
---|
64 | case 1: ret += changedyn0(n-5, 1); |
---|
65 | case 0: ret += changedyn0(n-2, 0); |
---|
66 | } |
---|
67 | cache[max][n] = ret; |
---|
68 | return ret; |
---|
69 | } |
---|
70 | |
---|
71 | long changedyn(int n, int max=4) { |
---|
72 | if(n<0) |
---|
73 | return 0; |
---|
74 | |
---|
75 | if(max<0) |
---|
76 | return 1; |
---|
77 | |
---|
78 | static std::unordered_map<int,long> cache[5]; |
---|
79 | std::unordered_map<int,long>::iterator it = cache[max].find(n); |
---|
80 | if(it != cache[max].end()) |
---|
81 | return cache[max][n]; |
---|
82 | |
---|
83 | long ret = 1; |
---|
84 | switch(max) { |
---|
85 | case 4: ret += changedyn(n-50, 4); |
---|
86 | case 3: ret += changedyn(n-20, 3); |
---|
87 | case 2: ret += changedyn(n-10, 2); |
---|
88 | case 1: ret += changedyn(n-5, 1); |
---|
89 | case 0: ret += changedyn(n-2, 0); |
---|
90 | } |
---|
91 | cache[max][n] = ret; |
---|
92 | return ret; |
---|
93 | } |
---|
94 | |
---|
95 | template<typename ST> |
---|
96 | stream<long> times(long n, long val, ST && s) |
---|
97 | { |
---|
98 | if(n==0) return s; |
---|
99 | return times(n-1, val, val<<=std::forward<ST>(s)); |
---|
100 | } |
---|
101 | |
---|
102 | stream<long> change1 = 1l<<=change1; |
---|
103 | stream<long> change2 = times(2, 0, change2) + change1; |
---|
104 | stream<long> change5 = times(5, 0, change5) + change2; |
---|
105 | stream<long> change10 = times(10, 0, change10) + change5; |
---|
106 | stream<long> change20 = times(20, 0, change20) + change10; |
---|
107 | stream<long> change50 = times(50, 0, change50) + change20; |
---|
108 | |
---|
109 | int main(int, char *argv[]) { |
---|
110 | // stream<long>::iterator it = change50.begin(); |
---|
111 | // for(int i=0; i<20; ++i) |
---|
112 | // { |
---|
113 | // assert(change(i) == *it); |
---|
114 | // std::cout<<i<<" "<<change(i)<<" "<<changef(i)<<" "<<changedyn(i)<<" "<<*it<<std::endl; |
---|
115 | // ++it; |
---|
116 | // } |
---|
117 | |
---|
118 | //std::cout<<j<<" "<<change(j)<<std::endl; |
---|
119 | //std::cout<<j<<" "<<changef(j)<<std::endl; |
---|
120 | |
---|
121 | const int n = atoi(argv[2]); |
---|
122 | const int x = atoi(argv[1]); |
---|
123 | |
---|
124 | struct timeval tv1, tv2; |
---|
125 | struct timezone tz; |
---|
126 | switch(x) { |
---|
127 | case 1: |
---|
128 | gettimeofday(&tv1, &tz); |
---|
129 | changedyn0(n); |
---|
130 | gettimeofday(&tv2, &tz); |
---|
131 | break; |
---|
132 | |
---|
133 | case 2: |
---|
134 | gettimeofday(&tv1, &tz); |
---|
135 | changedyn(n); |
---|
136 | gettimeofday(&tv2, &tz); |
---|
137 | break; |
---|
138 | |
---|
139 | case 3: |
---|
140 | gettimeofday(&tv1, &tz); |
---|
141 | stream<long>::iterator it = change50.begin(); |
---|
142 | for(int i=0; i<n; ++i) ++it; |
---|
143 | *it; |
---|
144 | gettimeofday(&tv2, &tz); |
---|
145 | break; |
---|
146 | } |
---|
147 | std::cout<<timeval_diff(tv2, tv1)<<std::endl; |
---|
148 | } |
---|