题目
题目地址:
牛客网--完美世界笔试原题
给定一个长度为N的数组,找出一个最长的单调自增子序列(不一定连续,但是顺序不能乱)
例如:给定一个长度为8的数组A{1,3,5,2,4,6,7,8},则其最长的单调递增子序列为{1,2,4,6,7,8},长度为6.
例如:给定一个长度为8的数组A{1,3,5,2,4,6,7,8},则其最长的单调递增子序列为{1,2,4,6,7,8},长度为6.
输入描述:
第一行包含一个整数T,代表测试数据组数。 对于每组测试数据: N-数组的长度 a1 a2 ... an (需要计算的数组) 保证: 1<=N<=3000,0<=ai<=MAX_INT.
输出描述:
对于每组数据,输出一个整数序列,代表最长递增子序列。 若有多组最长上升子序列,输出第一组。 保证:1<=T<=20,1<=N<=3000,0<=ai<=MAX_INT.
输入例子:
2 7 89 256 78 1 46 78 8 5 6 4 8 2 17
输出例子:
1 46 78 6 8 17
分析
首先得到每个元素位置对应的当前最长递增子序列数组。保存长度以及其上一个元素索引。
参考:
求最长递增子序列长度
代码
#include <iostream>
#include <cstdlib>
#include <vector>
#include <algorithm>
using namespace std;
struct Node {
int maxLen;
int lastIdx;
Node(int m, int l) :maxLen(m), lastIdx(l) {}
};
vector<int> maxSubSequence(vector<int> &arr)
{
if (arr.empty())
return vector<int>();
int n = arr.size();
vector<Node> dp(n, Node(0, 0));
dp[0].maxLen = 1;
dp[0].lastIdx = -1;
int maxLen = dp[0].maxLen;
int maxIdx = 0;
for (int i = 1; i<n; ++i)
{
dp[i].maxLen = 1;
dp[i].lastIdx = -1;
for (int j = 0; j<i; ++j)
{
if (arr[j] < arr[i])
{
if (dp[j].maxLen + 1 > dp[i].maxLen)
{
dp[i].maxLen = dp[j].maxLen + 1;
dp[i].lastIdx = j;
}
if (dp[i].maxLen > maxLen)
{
maxLen = dp[i].maxLen;
maxIdx = i;
}
}//if
}//for
}//for
//填充结果
vector<int> maxRet(maxLen, 0);
for (int i = maxIdx; maxLen >0 && i >= 0; )
{
maxRet[--maxLen] = arr[i];
i = dp[i].lastIdx;
}
return maxRet;
}
int main()
{
int n = 0;
cin >> n;
for (int i = 0; i<n; ++i)
{
int len = 0;
cin >> len;
vector<int> arr(len, 0);
for (int j = 0; j<len; ++j)
{
int tmp;
cin >> tmp;
arr[j] = tmp;
}//for
vector<int> ret = maxSubSequence(arr);
cout << ret[0];
for (unsigned int j = 1; j<ret.size(); ++j)
cout << " " << ret[j];
cout << endl;
}
system("pause");
return 0;
}