취미로 코딩하기

※ 글쓴이는 취미로 코딩을 익혀보는 사람이라 정확하지 않은 내용을 담고 있을 수 있다 ※

이번에 볼 문제는 백준 25289번 문제인 가장 긴 등차 부분 수열이다.
문제는 아래 링크를 확인하자.

https://www.acmicpc.net/problem/25289 

입력조건에 각 항은 1 이상 100 이하인 정수만 주어진다는 점을 눈여겨보자.

각 항이 1 이상 100 이하이므로, 가능한 공차는 -99 이상 99 이하의 정수들 뿐이다.

가능한 모든 각 공차에 대해 가능한 가장 긴 등차 부분 수열의 길이를 dp를 이용해 구하고, 그중 최댓값을 답으로 출력하자.

아래는 제출한 소스코드이다.

#include <iostream>
#include <cstring>
using namespace std;

int N;
int arr[100000];
int dp[101];

int main() {
	ios::sync_with_stdio(0);
	cin.tie(0);
	
	cin >> N;
	for (int i = 0; i < N; i++) cin >> arr[i];

	int ans = 0;

	for (int d = -99; d < 100; d++) {
		memset(dp, 0, sizeof(dp));

		for (int i = 0; i < N; i++) {
			int x = arr[i];
			if (0 < x - d && x - d < 101) dp[x] = max(dp[x], dp[x - d] + 1);
			else dp[x] = 1;
		}
		for (int i = 1; i < 101; i++) ans = max(ans, dp[i]);
	}

	cout << ans;
}