编程题 | 小马智行题目

题目描述： 

有N个石头，从第一个石头跳到第N个石头。每个石头的颜色为Ci，第i个石头往后跳只有两种跳法： 1. 往后跳一个； 2. 往后直接跳到相同颜色的石头上。  问： 从第一个石头跳到第N个石头，最少的步数是多少？   

输入： 第一行为N，第二行表示各石头的颜色  

6

1 2 4 1 4 2 

输出： 

2 （备注： 1->2->2）

---

解答： 

思路为递归函数，但是考虑到时间效率，需用循环结构从后往前实现。另外可用map<key, value>结构 计算第i个位置对应的下一相同颜色的位置，从而避免再用循环结构以提高效率。    

//
// Created by max on 2020/4/4.
//
#include <iostream>
#include <cmath>
#include <string>
#include <vector>
#include <cstdio>
#include <map>
using namespace std;

int main()
{
    int N;
    scanf("%d",&N);
    vector<int> Vec(N,0);
    vector<int> index(N, 0);  // 第i个位置对应的下一相同颜色位置
    map<int, int> m;
    int a;

    for(int i=0; i<N; ++i)
    {
        cin >> a;
        Vec[i] = a;
        if(m.find(a) == m.end())
        {
            m[a] = i;
        } else
        {
            int b = m[a];
            index[b] = i;
            m[a] = i;
        }
    }

    vector<int> ans(N,0);  // ans为递归函数
    ans[N-1] = 0;

    int start = N-2;
    while( start >= 0 )
    {
        int k = index[start];
        if(k == start + 1 || k == 0)
        {
            ans[start] = 1 + ans[start+1];
        }
        else
        {
            ans[start] = min( 1+ ans[start+1], 1+ ans[k] );
        }
        start--;
    }

    cout << ans[0] << endl;

    return 0;
}