PTA甲级 1045 Favorite Color Stripe (C++)

Eva is trying to make her own color stripe out of a given one. She would like to keep only her favorite colors in her favorite order by cutting off those unwanted pieces and sewing the remaining parts together to form her favorite color stripe.

It is said that a normal human eye can distinguish about less than 200 different colors, so Eva’s favorite colors are limited. However the original stripe could be very long, and Eva would like to have the remaining favorite stripe with the maximum length. So she needs your help to find her the best result.

Note that the solution might not be unique, but you only have to tell her the maximum length. For example, given a stripe of colors {2 2 4 1 5 5 6 3 1 1 5 6}. If Eva’s favorite colors are given in her favorite order as {2 3 1 5 6}, then she has 4 possible best solutions {2 2 1 1 1 5 6}, {2 2 1 5 5 5 6}, {2 2 1 5 5 6 6}, and {2 2 3 1 1 5 6}.

Input Specification:

Each input file contains one test case. For each case, the first line contains a positive integer $N(\le 200)$ which is the total number of colors involved (and hence the colors are numbered from 1 to $N$). Then the next line starts with a positive integer $M(\le 200)$ followed by $M$ Eva’s favorite color numbers given in her favorite order. Finally the third line starts with a positive integer $L(\le 10^4)$ which is the length of the given stripe, followed by $L$ colors on the stripe. All the numbers in a line a separated by a space.

Output Specification:

For each test case, simply print in a line the maximum length of Eva’s favorite stripe.

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

Caution:

可以转化成 Leetcode 上这道最长递增子序列的问题。

Solution:

// Talk is cheap, show me the code
// Created by Misdirection 2021-08-24 13:32:54
// All rights reserved.

#include <iostream>
#include <vector>
#include <algorithm>

using namespace std;

int main(){
    int n, m, l;
    // n - the total number of colors involved
    // followed by M Eva's favorite color numbers given in her favorite order

    scanf("%d", &n);
    scanf("%d", &m);
    vector<int> eva(m);
    vector<vector<int>> position(m);
    for(int i = 0; i < m; ++i) scanf("%d", &eva[i]);

    scanf("%d", &l);
    int size = 0;

    for(int i = 0; i < l; ++i){
        int tmp;
        scanf("%d", &tmp);

        int pos = find(eva.begin(), eva.end(), tmp) - eva.begin();

        if(pos != m){
            position[pos].push_back(i);
            size++;
        }
    }

    int* tmp = new int[size];
    int pos = 0;
    for(int i = 0; i < m; ++i)
        for(int j = 0; j < position[i].size(); ++j) tmp[pos++] = position[i][j];

    int* dp = new int[size];
    for(int i = 0; i < size; ++i){
        dp[i] = 1;

        for(int j = 0; j < i; ++j){
            if(tmp[j] < tmp[i]) dp[i] = max(dp[i], dp[j] + 1);
        }
    }
    
    printf("%d\n", *max_element(dp, dp + size));

    return 0;
}