首页 > 代码库 > Codility上的练习(11)
Codility上的练习(11)
(1)Ladder
给定两个等长的数组A和B, A[i]和B[i]表示求一个有A[i]级的梯子,每次上1级或者两级,上到最高级的方法数对2^B[i]取余数的结果。
数据范围:数组长度 L [1..30000] , A中数字范围 [1..L], B中数字范围[1..30]
要求复杂度 时间空间都是O(L)
分析:打表法——我们循环可以把0..L的结果都算出来 f[i] = f[i - 1] + f[i - 2] , f[0] = f[1] = 1。 关键点在于取余数,这个取余数很特殊,对2^B[i]取余数,相当于取结果的低B[i] bit。因此我们可以取所有结果的低30位,打好表,输出时再真正取B[i]位,就能达到O(L)的复杂度了。
// you can also use includes, for example: // #include <algorithm> const int M = (1 << 30) - 1; vector<int> solution(vector<int> &A, vector<int> &B) { // write your code in C++98 int m = 0; for (int i = 0; i < A.size(); ++i) { m = max(m, A[i]); } vector<int> f; f.resize(m + 1); f[0] = f[1] = 1; for (int i = 2; i <= m; ++i) { f[i] = (f[i - 1] + f[i - 2]) & M; } vector<int> answer; for (int i = 0; i < A.size(); ++i) { answer.push_back(f[A[i]] & ((1 << B[i]) - 1)); } return answer; }
(2) FibFrog
先给了Fibonacci的定义F[0] = 0, F[1] = 1, F(M) = F(M - 1) + F(M - 2) if M >= 2。一只青蛙想从河岸(-1)跳到对岸(N)。中间有0..N-1这N个位置,有一个数组A[]表示这些位置是否有荷叶 ,0表示没有,1表示有。青蛙从岸上要跳到荷叶上,通过0个或者多个荷叶跳到对岸,而且只能朝对岸的方向跳——不能往回跳。而且青蛙每次跳的距离必须是一个Fibonacci数,请问青蛙最少几步跳到对岸?无解返回-1。
数据范围: N [0..30000]
要求复杂度 :时间O(NlogN),空间O(N)。
分析: 首先我们可以把数组尾端加一个1,表示对岸。可以认为首端有一个1或者认为数组下标从1开始,我们现在目标是到数组最后一个元素,只能通过1且跳的距离是Fibonacci数的最小步数。 这是一个显然的dp (或者说是bfs)。对于位置i 并且A[i] == 1,我们有 dp[i] = min(dp[i - f[j]]) + 1 其中f[j]是fibonacci数并且满足A[i - f[j]]==1
因为Fibonacci是指数增长的,所以不超过N的Fibonacci数是O(logN)个,整个递推的时间复杂度是O(NlogN)。
// you can also use includes, for example: // #include <algorithm> int solution(vector<int> &A) { // write your code in C++98 A.push_back(1); int n = A.size(); vector<int> f; f.push_back(1); f.push_back(1); for (int i = 1; f[i] < n; ++i) { f.push_back(f[i] + f[i - 1]); } vector<int> answer; answer.resize(n + 1, -1); answer[0] = 0; for (int i = 1; i <= n; ++i) { answer[i] = -1; if (A[i - 1] == 0) { continue; } for (int j = 0;(j < f.size()) && (i >= f[j]); ++j) { if ((answer[i - f[j]] >= 0) && ((answer[i] < 0) || (answer[i] > answer[i - f[j]] + 1))) { answer[i] = answer[i - f[j]] + 1; } } } return answer[n]; }
Codility上的练习(11)