PAT甲级1045 Favorite Color Stripe (30分)|C++实现

一、题目描述

原题链接

Input Specification:

​​Output Specification:

Sample Input:

6
5 2 3 1 5 6
12 2 2 4 1 5 5 6 3 1 1 5 6

Sample Output:

7

二、解题思路

一道动态规划的题目。我们可以把输入数据中无用的颜色滤掉，即只把喜欢的颜色存入一个vector，同时设置一个存放优先级的数组，优先级越高，对应数字越大，排得越靠前。dp[i]表示以第i个数结尾的颜色序列的最大长度，对于每一个store中的每一个元素，我们都比较这个颜色和前一个颜色的优先级，如果当前颜色优先级比较低，则加入序列，否则我们暂时将这个dp[i]设置为1，随后计算前面第一个优先级比当前颜色大的颜色，对应的dp[k]+1，我们取dp[i]和dp[k]+1的最大值，此即为状态转移方程。具体实现详见注释。

三、AC代码

#include<iostream>
#include<cstdio>
#include<vector>
#include<algorithm>
using namespace std;
const int maxf = 201;   //喜欢的颜色的最大数目
const int maxn = 10001; //最大值
int order[maxf] = {0};  //优先级，优先级大则可在前面
bool in_fav[maxn] = {false};    //颜色是否在喜欢的颜色之列
int dp[maxn] = {0}; //动态规划数组
vector<int>	store;  //题目给出的颜色中，在喜欢的颜色之列的所有颜色
int main()
{
  int N, L, F, ans=0, tmp;
  scanf("%d", &N);
  scanf("%d", &F);
  for(int i=0; i<F; i++)
  {
    scanf("%d", &tmp);
    in_fav[tmp] = true;
    order[tmp] = maxf-i;    //设置优先级，优先级高的在前面
  }
  scanf("%d", &L);
  for(int i=0; i<L; i++)
  {
    scanf("%d", &tmp);
    if(in_fav[tmp])	store.push_back(tmp);   //喜欢的颜色存入store
  }
  dp[0] = 1;    //至少有一个颜色
  ans = dp[0];
  int sze = store.size();
  for(int i=1; i<sze; i++)
  {
    if(order[store[i]] <= order[store[i-1]])	dp[i] = dp[i-1] + 1;    //如果当前的优先级比前面的低，则可以放入
    else dp[i] = 1;
    int k=i-1;    //k取前面一个数
    while(k>=0 && order[store[i]] > order[store[k]])  k--;    //找出第i个数之前第一个优先级比k大的数
    if(k < 0);
    else dp[i] = max(dp[i], dp[k] + 1);
    if(dp[i] > ans)	ans = dp[i];
  }
  printf("%d", ans);
  return 0;
}