읽는데 약 2분
[백준 1796] 신기한 키보드
문제
https://www.acmicpc.net/problem/1796
알고리즘
DP
풀이
$cache[i][j]$ = $i$번째 알파벳부터 해결해야 하고,현재 $idx$위치가 $j$ 일 때 필요한 최소 이동 횟수
현재 커서 위치에서 다음 단계의 알파벳으로 가기 위해선 현재 해결해야하는 알파벳의 $lo$ 위치와 $hi$ 위치를 들러야 합니다. 두 위치를 들른다면, 그 사이에 있는 현재 단계의 알파벳은 모두 해결 가능함을 알 수 있습니다.
$dist(a,b,c,d)$= 현재 위치 $a$에서 $c$와 $d$를 들른 후 $b$까지 가기 위해 필요한 커서 거리
$dist$함수를 통해 현재 단계에서 필요한 최소 커서 이동거리를 구할 수 있고, 다음 단계로 나아가 문자$z$까지 처리를 해주면 정답을 얻을 수 있습니다.
코드
#include <bits/stdc++.h>
#define rep(i,n) for(int i=0;i<n;++i)
#define REP(i,n) for(int i=1;i<=n;++i)
#define FAST cin.tie(NULL);cout.tie(NULL); ios::sync_with_stdio(false)
using namespace std;
string S;
int s;
int cache[26][1001];
int lo[27],hi[27];
bool exist[26];
int dist(int a, int b, int c,int d) {
if (c == -1 || d == -1) return 0;
int A = abs(a - c);
int B = abs(c - d);
int C = abs(d - b);
return A + B + C;
}
int solve(int alpha, int pos) {
if (alpha==26) return 0;
int& ret = cache[alpha][pos];
if (ret != -1) return ret;
ret = INT_MAX/2;
int posl = lo[alpha];
int posr = hi[alpha];
if (exist[alpha])
rep(i, s)
ret = min(ret, solve(alpha + 1, i) + min(dist(pos, i, posl, posr), dist(pos, i, posr, posl)));
else ret =solve(alpha + 1, pos);
return ret;
}
int main() {
FAST;
cin >> S;
s = S.size();
memset(cache, -1, sizeof(cache));
memset(lo, 0x3f, sizeof(lo));
memset(hi, -1, sizeof(hi));
rep(i, 26) rep(j, s) {
if (S[j] - 'a' == i) {
exist[i] = 1;
lo[i] = min(lo[i], j);
hi[i] = max(hi[i], j);
}
}
int ret = solve(0, 0);
cout << ret + s;
return 0;
}
Read other posts