Querying the Document

  • + 0 comments

    We can use strtok_r and save its internal states using save pointers since we use nested looping.

    #include <stdio.h>
    #include <stdlib.h>
    #include <string.h>
    #include<assert.h>
    #define MAX_CHARACTERS 1005
    #define MAX_PARAGRAPHS 5
    
    int char_count(char *s, char c) {
        int count = 0;
        for (int i = 0; i < strlen(s); i++)
            count += (int) (s[i] == c);
        return count;
    }
    
    char* kth_word_in_mth_sentence_of_nth_paragraph(char**** document, int k, int m, int n) {
        return document[n - 1][m - 1][k - 1];
    }
    
    char** kth_sentence_in_mth_paragraph(char**** document, int k, int m) {
        return document[m - 1][k - 1];
    }
    
    char*** kth_paragraph(char**** document, int k) {
        return document[k - 1];
    }
    
    char**** get_document(char* text) {
        char ****doc;
        int pcnt = char_count(text, '\n') + 1, cp = 0;
        doc = (char ****)malloc((sizeof(char ***)) * pcnt);
        char *ptoken, *psaveptr;
        ptoken = strtok_r(text, "\n", &psaveptr);
        while (ptoken != NULL) {
            char *p = (char *)malloc((sizeof(char)) * (strlen(ptoken) + 1));
            strcpy(p, ptoken);
            int scnt = char_count(p, '.') + 1, cs = 0;
            doc[cp] = (char ***)malloc((sizeof(char **)) * scnt);
            char *stoken, *ssaveptr;
            stoken = strtok_r(p, ".", &ssaveptr);
            while (stoken != NULL) {
                char *s = (char *)malloc((sizeof(char)) * (strlen(stoken) + 1));
                strcpy(s, stoken);
                int wcnt = char_count(s, ' ') + 1, cw = 0;
                doc[cp][cs] = (char **)malloc((sizeof(char *)) * wcnt);
                char *wtoken, *wsaveptr;
                wtoken = strtok_r(s, " ", &wsaveptr);
                while (wtoken != NULL) {
                    char *w = malloc((sizeof(char)) * (strlen(wtoken) + 1));
                    strcpy(w, wtoken);
                    int ccnt = strlen(w);
                    doc[cp][cs][cw] = (char *)malloc((sizeof(char)) * (ccnt + 1));
                    strcpy(doc[cp][cs][cw], w);
                    wtoken = strtok_r(NULL, " ", &wsaveptr);
                    cw++;
                }
                stoken = strtok_r(NULL, ".", &ssaveptr);
                cs++;
            }
            ptoken = strtok_r(NULL, "\n", &psaveptr);
            cp++;
        }
        return doc;
    }
    
    
    
    char* get_input_text() {	
        int paragraph_count;
        scanf("%d", &paragraph_count);
    
        char p[MAX_PARAGRAPHS][MAX_CHARACTERS], doc[MAX_CHARACTERS];
        memset(doc, 0, sizeof(doc));
        getchar();
        for (int i = 0; i < paragraph_count; i++) {
            scanf("%[^\n]%*c", p[i]);
            strcat(doc, p[i]);
            if (i != paragraph_count - 1)
                strcat(doc, "\n");
        }
    
        char* returnDoc = (char*)malloc((strlen (doc)+1) * (sizeof(char)));
        strcpy(returnDoc, doc);
        return returnDoc;
    }
    
    void print_word(char* word) {
        printf("%s", word);
    }
    
    void print_sentence(char** sentence) {
        int word_count;
        scanf("%d", &word_count);
        for(int i = 0; i < word_count; i++){
            printf("%s", sentence[i]);
            if( i != word_count - 1)
                printf(" ");
        }
    } 
    
    void print_paragraph(char*** paragraph) {
        int sentence_count;
        scanf("%d", &sentence_count);
        for (int i = 0; i < sentence_count; i++) {
            print_sentence(*(paragraph + i));
            printf(".");
        }
    }
    
    int main() 
    {
        char* text = get_input_text();
        char**** document = get_document(text);
    
        int q;
        scanf("%d", &q);
    
        while (q--) {
            int type;
            scanf("%d", &type);
    
            if (type == 3){
                int k, m, n;
                scanf("%d %d %d", &k, &m, &n);
                char* word = kth_word_in_mth_sentence_of_nth_paragraph(document, k, m, n);
                print_word(word);
            }
    
            else if (type == 2){
                int k, m;
                scanf("%d %d", &k, &m);
                char** sentence = kth_sentence_in_mth_paragraph(document, k, m);
                print_sentence(sentence);
            }
    
            else{
                int k;
                scanf("%d", &k);
                char*** paragraph = kth_paragraph(document, k);
                print_paragraph(paragraph);
            }
            printf("\n");
        }     
    }