LINTCODE——Folding Array

LINTCODE——Folding Array

给一个长度为 n 的数组 nums 和一个长度为 k 的数组 req, 你需要根据要求折叠数组，并输出折叠的结果
1.如果 req[i] = 0 意味着你应该从左到右折叠, 例如:

1 2 3 4 5 6 7 8 ==> 4 3 2 1
5 6 7 8
2.如果 req[i] = 1 意味着你应该从右往左折叠, 例如:

1 2 3 4 5 6 7 8 ==> 8 7 6 5
1 2 3 4
更多的例子:

从左到右折叠
4 3 2 1 ==> 6 5
5 6 7 8 3 4
2 1
7 8

从右到左折叠
6 5 ==> 8
3 4 1
2 1 4
7 8 5
6
3
2
7

分析值：左折叠时，右不变、下移,左翻转、位于头部；右折叠，左不变、下移，右翻转，位于头部；ps：代码写的不太好，不过主要是这个思路，左翻右翻；

class Solution {
public:
    /*
     * @param : the original array
     * @param : the direction each time
     * @return: the final folded array 
     */
    vector<int> folding(vector<int> &nums, vector<int> &req) {
        // write your code here

        for(int i = 0; i < req.size(); i++)
            if(req[i] == 0)
                left(nums, pow(2,i));
            else
                right(nums, pow(2, i));

        return nums;

    }
    void left(vector<int> &nums, int index)
    {
        //左折叠，右不变、下移,左翻转、位于头部
        int cnt = nums.size()/(index*2);
        vector<int> out;
        for(int i = 0; i < index; i++)
        {
            //迭代器末尾的位置，等于开头的+cnt；
            vector<int> temp1(nums.begin()+i*cnt*2, nums.begin()+i*cnt*2 + cnt);
            vector<int> temp2(nums.begin()+i*cnt*2+cnt, nums.begin()+i*cnt*2+cnt + cnt);

            reverse(temp1.begin(), temp1.end());

            out.insert(out.begin(), temp1.begin(), temp1.end());
            out.insert(out.end(), temp2.begin(), temp2.end());
        }
        nums = out;
        return;
    }
    void right(vector<int> &nums, int index)
    {
        //右折叠，左不变、下移，右翻转，位于头部
        int cnt = nums.size()/(index*2);
        vector<int> out;
        for(int i = 0; i < index; i++)
        {
            vector<int> temp1(nums.begin()+i*cnt*2, nums.begin()+i*cnt*2 + cnt);
            vector<int> temp2(nums.begin()+i*cnt*2+cnt, nums.begin()+i*cnt*2+cnt + cnt);

            reverse(temp2.begin(), temp2.end());

            out.insert(out.begin(), temp2.begin(), temp2.end());
            out.insert(out.end(), temp1.begin(), temp1.end());
        }
        nums = out;
        return;

    }
};