https://www.acmicpc.net/problem/5592
5592번: 最長の階段
n = 7, k = 5 とする．6, 2, 4, 7 と白紙のカードが与えられたとき，このカードを 使って作れる連続した整数列のうち最長のものは，白紙のカードに 5 を書くことに よってできる 4, 5, 6, 7 であり
www.acmicpc.net
0이 포함되지 않은 경우와 0이 포함된 경우로 나눈다.
0이 포함되지 않은 경우는 매우 쉽다.
0이 포함된 경우가 지랄이다.
투 포인터를 쓰자.
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
|
#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;
int main()
{
int n, k;
cin >> n >> k;
vector<int> arr(k);
bool flag = false;
for (int i = 0; i < k; i++)
{
cin >> arr[i];
if (arr[i] == 0)
flag = true;
}
sort(arr.begin(), arr.end());
int ans = 1;
if (flag)
{
// 그 숫자가 존재한다면 true
vector<bool> exist(n + 1);
for (int i = 0; i < k; i++)
{
exist[arr[i]] = true;
}
//투포인터를 쓴다
int l = 0, r = 0;
// 1부터 n까지를 전부 조사
for (int i = 1; i <= n; i++)
{
if (exist[i])
{
r++;
}
// 존재하지 않는다? 그럼 불연속이 발생
else
{
ans = max(ans, l + r + 1);
l = r;
r = 0;
}
}
ans = max(ans, l + r + 1);
}
else
{
int tmp = 1;
for (int i = 0; i < k - 1; i++)
{
if (arr[i] + 1 == arr[i + 1])
{
tmp++;
}
else
{
ans = max(ans, tmp);
tmp = 1;
}
}
ans = max(ans, tmp);
}
cout << ans << '\n';
return 0;
}
|
cs |