BUPT-homework/semester2/challenge1/main.c

#include <stdio.h>
#include <string.h>
#include <stdlib.h>

#define NUM_USERS 611
#define NUM_MOVIES 9743
#define NUM_GENRES 20

typedef struct Movies
{
    int movieID;
    struct Movies *next;
} Movies;

typedef struct Genres
{
    int genreID;
    struct Genres *next;
} Genres;

//输入电影风格返回其映射的id
int movie_style_map(char *style)
{

    int index = 0;
    const char *movies[] = {"Action", "Adventure", "Animation", "Children's",
                            "Comedy", "Crime", "Documentary", "Drama",
                            "Fantasy", "Film-Noir", "Horror", "IMAX", "Musical",
                            "Mystery", "Romance", "Sci-Fi", "Thriller",
                            "War", "Western"};

    while (strcmp(movies[index], style) != 0)
    {
        index++;
    }
    return index;
}

// 函数原型
Movies *appendMovies(Movies *last, int id);
int getGenreID(int movieID, Genres ** movies); // 传入movieID,输出风格下标
void clearArrInt(int* arr, int size); // 将所有元素设为0

int main(void)
{
    // Firstly open the ratings.csv
    FILE *fp;
    fp = fopen("./ratings.csv", "r");

    char *buf = malloc(256 * sizeof(char));

    // Declare the users table and init with null.
    Movies *users[NUM_USERS];
    for (int i = 0; i < NUM_USERS; i++)
    {
        users[i] = NULL;
    }

    fgets(buf, 256 * sizeof(char), fp);

    // 看每个用户喜欢什么电影
    int userID, movieID;
    char input[64];
    int loc, loc1;
    while (fgets(buf, 256 * sizeof(char), fp) != NULL)
    {
        loc = 0;
        input[0] = '\0'; // clear the array
        // get userID
        while (buf[loc] != ',')
        {
            input[loc] = buf[loc];
            loc++;
        }
        input[loc++] = '\0';
        userID = atoi(input);
        // printf("UserID: %d\n", userID);

        // get movieID
        loc1 = 0;
        input[0] = '\0';
        while (buf[loc] != ',')
        {
            input[loc1++] = buf[loc];
            loc++;
        }
        input[loc1] = '\0';
        movieID = atoi(input);
        // printf("movieID: %d\n", movieID);
        users[userID] = appendMovies(users[userID], movieID);
    }

    fclose(fp);

    fp = fopen("./movies.csv", 'r');

    // 下标表示电影id
    Genres *movies[NUM_MOVIES];


    // TODO: 仿照我的代码填充*movies数组
    // 李王子写 ok 你写吧
    //

    //
    for(int i=0;i<NUM_MOVIES;i++)
    {
        movies[i]=NULL;
    }


    char*str=(char*)malloc(256*sizeof(char));

    //读第一行
    fgets(str, 256 * sizeof(char), fp); 
    
    //        fgets() returns s on success, and NULL on error or when end  of  file
    //   occurs while no characters have been read.
    //      你那个该写!=NULL 你看注释
    //  ok 

    
    int id_index=0;

    while(fgets(str,256*sizeof(char),fp)!=NULL)
    {
        

    }



    fclose(fp);

    // 开始输出答案到文件answer.txt

    fp = fopen("./answer.txt", 'w+'); // w+: 如果文件已存在先清空

    int count[NUM_GENRES] = {};

    // 开始从第一个用户循环 (1)
    Movies *ptr;
    for (int i = 1; i < NUM_GENRES; i++) {
        cleararrInt(count, NUM_GENRES);
        // 便历他喜欢的电影
        ptr = users[i];
        while (ptr != NULL) {
            count[getGenreID(ptr->movieID, movies)] ++;
            ptr = ptr->next;
        }
    }

    //那个 就是你帮我写的那个数组 我感觉开个一维数组就可以了啊 

    fclose(fp);

    return 0;
}

Movies *appendMovies(Movies *last, int id)
{
    Movies *new = malloc(sizeof(Movies));

    new->movieID = id;
    new->next = NULL;

    // 1.如果是null
    if (last == NULL)
    {
        return new;
    }
    // 2.不是null，在最后加上新节点
    Movies *ptr = last;

    while (ptr->next != NULL)
    {
        ptr = ptr->next;
    }

    ptr->next = new;

    return last;
}


int getGenreID(int movieID, Genres **movies) {
    Genres *ptr = movies[movieID];
}

void clearArrInt(int* arr, int size) {
    for (int i = 0; i < size; i++) {
        arr[i] = 0;
    }
}