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;
}