算法趣题-Q15

一、问题描述

二、问题分析

        A，B两个人的运动规则相同，那么先只考虑A或者B，此时就是一个基础的动态规划上台阶问题。那么当考虑A，B相向而行，且最后同时到达同一级台阶上的可能情况就是：同样步数时，A和B加起来一共走10级的可能性。而A，B运动规则相同则可以只计算其中一个人的情况数量，在根据题目进行阶梯数量对应相乘累加即可。在实现方面，可以对动态规划进行优化减少重复运算。

三、代码实现

1.C/C++实现

#include <iostream>
#define NUM 10

using namespace std;

int steps[5][NUM - 1];

int main()
{
    memset(steps, 0, sizeof(steps));
    for (int i = 0; i < NUM - 1; i++)
    {
        if (i < 4)
            steps[0][i] = 1;
        else
            steps[0][i] = 0;
    }

    for (int i = 1; i < 5; i++)
        for (int j = 0; j < NUM - 1; j++)
            for (int k = 1; k < 5; k++)
                if (j + k < NUM - 1)
                    steps[i][j + k] += steps[i - 1][j];

    int count = 0;
    for (int i = 0; i < 5; i++)
        for (int j = 0; j < NUM - 1; j++)
            count += steps[i][j] * steps[i][NUM - j - 2];

    cout << count << endl;
}

2.Python实现

# coding=utf-8

import math

# NUM 阶梯数目
NUM = 10
# step[i]存放后由下开始，走第i+1步后的对应阶梯的可能情况
step = []


def generate():
    step.append([1] * 4)
    step[0].extend([0] * (NUM - 4 - 1))
    for i in range(math.ceil(NUM / 2) - 1):
        # tmp[j]存放第i+1步后，能有多少种情况可以到第j级阶梯
        tmp = [0] * NUM
        for j in [1, 2, 3, 4]:
            for k in range(NUM - 1):
                if k + j < NUM - 1:
                    tmp[k + j] += step[i][k]
        step.append(tmp)


if __name__ == '__main__':
    generate()
    count = 0
    for i in range(math.ceil(NUM / 2)):
        for j in range(NUM - 1):
            count += step[i][j] * step[i][NUM - j - 2]
    print(count)