[算法题]孩子们的游戏(圆圈中最后剩下的数)

题目链接:孩子们的游戏(圆圈中最后剩下的数)

根据题目描述得知是从n个小朋友中找出最后剩下的孩子的编号(从0开始), 用动态规划来求解, 先定义状态表示:

dp[i]: 表示在 i 个孩子中最后剩下的孩子的编号

图示:

根据动态规划算法的规律, 状态转移方程一般为 dp[i] = dp[i-1] 和 dp[i-2] 怎样怎样, 而根据状态表示dp[i-1] 就是在 i-1 个孩子中最后剩下的孩子的编号, 可以得知最后剩下一个小孩子必然是每次dp[i], dp[i-1], dp[i-2]......所求的同一个, 假设图上的编号 2 是剩于 i-1 个孩子时最后剩下孩子的编号, 那么剩于 i 个孩子时最后剩下孩子的编号就等于 2+m, 此时的值需要模上当前剩于的孩子的个数, 因为它是按一个 "环" 来走的, 如果不进行取模会出现值不对的情况, 比如 n=5, m=3, 假如 n-m 是剩于 i-1 个孩子时最后剩下孩子的编号, 那么 n-m+m = 5-3+3 = 5, 编号对不上, 进行取模后 5%5 = 0 就对了, 因此状态转移方程为:

dp[i] = (dp[i-1] + m) % i

在题解时不需要一定创建一个dp数组, 因为状态转移只依赖 dp[i] 的前一项, 所以通过一个变量保存前一项即可, 题解代码如下:

class Solution 
{
public:
    int LastRemaining_Solution(int n, int m) 
    {
        //dp[i]表示当前环中存在i个孩子,i个孩子中最后获胜的下标
        int dp = 0; //纪录前一项
        int res = 0; //记录最终编号
        //当只有一个孩子时编号就为dp的初始值0,循环从两个孩子开始,一直到n个孩子
        for(int i = 2; i <= n; ++i)
        {
             res = (dp + m) % i;
             dp = res;
        }
        return res;
    }
};