[ boj : 22251 ] 빌런 호석

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

22251번: 빌런 호석

해설 : 이건 구현 문제입니다.

미리 숫자들간 차이값을 전처리로 구해줍시다.

 

그리고 난 다음에 최대 6자리이니 6칸짜리 배열 2개를 이용해서 각 자리 숫자들을 비교해 주는 것 입니다.

 

이때, -1로 초기화를 해놓고 K자리수 만큼만 숫자들을 하나씩 할당해 놓으면 됩니다.

 

그 다음 미리 전처리가 되어있는 숫자들을 사용해서 답을 구해주면 됩니다.

 

#include<bits/stdc++.h>

using namespace std;

int el[10][10];
int num[6];

void solve();
int get_ans(int x, int P, int K);
void init();

int main()
{
    int t = 1;
    //cin>>t;
    while (t--)
        solve();
    return 0;
}
void solve()
{
    init();
    int N, K, P, X;

    cin >> N >> K >> P >> X;

    memset(num, -1, sizeof(num));
    int x = X;
    for (int i = 0; i < K; i++) {
        num[i] = x % 10;
        x /= 10;
    }
    int ans = 0;
    for (int i = 1; i <= N; i++) {
        if (i == X)continue;
        ans += get_ans(i, P, K);
    }
    cout << ans;
}
int get_ans(int x, int P, int K)
{
    int num2[6];
    memset(num2, -1, sizeof(num2));
    for (int i = 0; i < K; i++) {
        num2[i] = x % 10;
        x /= 10;
    }
    for (int i = 0; i < K; i++) {
        P -= el[num[i]][num2[i]];
    }
    if (P >= 0)return 1;
    else return 0;
}
void init()
{
    el[0][1] = el[1][0] = 4;
    el[0][2] = el[2][0] = 3;
    el[0][3] = el[3][0] = 3;
    el[0][4] = el[4][0] = 4;
    el[0][5] = el[5][0] = 3;
    el[0][6] = el[6][0] = 2;
    el[0][7] = el[7][0] = 3;
    el[0][8] = el[8][0] = 1;
    el[0][9] = el[9][0] = 2;

    el[1][2] = el[2][1] = 5;
    el[1][3] = el[3][1] = 3;
    el[1][4] = el[4][1] = 2;
    el[1][5] = el[5][1] = 5;
    el[1][6] = el[6][1] = 6;
    el[1][7] = el[7][1] = 1;
    el[1][8] = el[8][1] = 5;
    el[1][9] = el[9][1] = 4;

    el[2][3] = el[3][2] = 2;
    el[2][4] = el[4][2] = 5;
    el[2][5] = el[5][2] = 4;
    el[2][6] = el[6][2] = 3;
    el[2][7] = el[7][2] = 4;
    el[2][8] = el[8][2] = 2;
    el[2][9] = el[9][2] = 3;

    el[3][4] = el[4][3] = 3;
    el[3][5] = el[5][3] = 2;
    el[3][6] = el[6][3] = 3;
    el[3][7] = el[7][3] = 2;
    el[3][8] = el[8][3] = 2;
    el[3][9] = el[9][3] = 1;

    el[4][5] = el[5][4] = 3;
    el[4][6] = el[6][4] = 4;
    el[4][7] = el[7][4] = 3;
    el[4][8] = el[8][4] = 3;
    el[4][9] = el[9][4] = 2;

    el[5][6] = el[6][5] = 1;
    el[5][7] = el[7][5] = 4;
    el[5][8] = el[8][5] = 2;
    el[5][9] = el[9][5] = 1;

    el[6][7] = el[7][6] = 5;
    el[6][8] = el[8][6] = 1;
    el[6][9] = el[9][6] = 2;

    el[7][8] = el[8][7] = 4;
    el[7][9] = el[9][7] = 3;

    el[8][9] = el[9][8] = 1;
}