N个不同的正整数,找出由这些数组成的最长的等差数列。
例如:1 3 5 6 8 9 10 12 13 14
等差子数列包括(仅包括两项的不列举)
1 3 5
1 5 9 13
3 6 9 12
3 8 13
5 9 13
6 8 10 12 14
其中6 8 10 12 14最长,长度为5。
Input
第1行:N,N为正整数的数量(3 <= N <= 10000)。
第2 - N+1行:N个正整数。(2<= A[i] <= 10^9)
Output
最长等差数列的长度。
Input示例
10
1
3
5
6
8
9
10
12
13
14
Output示例
5
思路:二维动态规划的变种,从后往前遍历,依次查找最长等差数列。
#include <iostream>
#include <string>
#include <string.h>
#include <vector>
#include <algorithm>
#include <iostream>
#include <set>
#include <algorithm>
#include <sstream>
#include <queue>
#include <iomanip>
using namespace std;
int main()
{
int N;
vector<int> data;
int tmp;
int dp[100][100];
vector<int> res;
cin >> N;
for (int i = 0; i < N; i++)
{
cin >> tmp;
data.push_back(tmp);
}
sort(data.begin(), data.end());
for (int i = 0; i < data.size(); i++)
{
for (int j = i + 1; j < data.size(); j++)
{
dp[i][j] = 2;
}
}
int result = 2;
int max_i = -1;
int max_j = -1;
for (int j = data.size() - 2; j >= 1; j--) //注意下标的起始和结束
{
int i = j - 1;
int k = j + 1;
while (i >= 0 && k < data.size())
{
int left = data[j] - data[i]; //由当前元素向两侧展开搜索
int right = data[k] - data[j];
if (left < right)
{
i--;
} else if (left > right)
{
k++;
} else {
dp[i][j] = dp[j][k] + 1;
if (dp[i][j] > result)
{
result = dp[i][j];
max_i = i; //记录最长等差数列第一个数的下标
max_j = j; //记录最长等差数列第二个数的下标
}
i--;
k++;
}
}
}
//存储最长等差数列(根据等差数列第一项和差值反向查找)
res.push_back(data[max_i]);
int last = max_i;
for (int i = max_j; i < data.size(); i++)
{
if (data[i] - data[last] == data[max_j] - data[max_i])
{
last = i;
res.push_back(data[i]);
}
}
cout << result << endl;
return 0;
}
参考:
[1] https://blog.csdn.net/brucehb/article/details/79285585
[2] https://blog.csdn.net/rabbitsockx/article/details/84307701