%{
#include <stdio.h>
#include <string.h>

// List of keywords
char *keywords[] = {
    "int", "float", "char", "double", "if", "else",
    "while", "for", "return", "void", "break", "continue"
};

int isKeyword(char *word) {
    int i;
    for(i = 0; i < 12; i++) {
        if(strcmp(word, keywords[i]) == 0)
            return 1;
    }
    return 0;
}
%}

%%
[a-zA-Z_][a-zA-Z0-9_]* {
    if(isKeyword(yytext))
        printf("%s is a KEYWORD\n", yytext);
    else
        printf("%s is an IDENTIFIER\n", yytext);
}

\n      { return 0; }
.       ;
%%

int main() {
    printf("Enter a word: ");
    yylex();
    return 0;
}

int yywrap() {
    return 1;
}