一、题目描述
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;
}