DXライブラリでキーが押された瞬間を判別する

掲題の通りです。
質問があったもので

#include "DxLib.h"

// キー入力
int key[256]; // 0:入力されていない 1:入力された瞬間 2:入力されている
void inputKey() {
    static char buf[256];
    GetHitKeyStateAll(buf);
    for (int i = 0; i < 256; i++) {
        if (buf[i]) {
            if (key[i] == 0) key[i] = 1;
            else if (key[i] == 1) key[i] = 2;
        } else key[i] = 0;
    }
}

int WINAPI WinMain(HINSTANCE, HINSTANCE, LPSTR, int) {
    ChangeWindowMode(TRUE);
    DxLib_Init();
    SetDrawScreen(DX_SCREEN_BACK);
    while (!ProcessMessage()) {
        ClearDrawScreen();
        inputKey();
        if (key[KEY_INPUT_SPACE] == 1) {
            DrawString(0, 0, "SPACE押された瞬間", GetColor(255, 255, 255));
        } else if (key[KEY_INPUT_SPACE] == 2) {
            DrawString(0, 24, "SPACE押されている", GetColor(255, 255, 255));
        }
        ScreenFlip();
    }
    DxLib_End();
    return 0;
}