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

typedef struct {
    int employee_id;
    int project_id;
    int workload;
    char name[50];
    char team[50];
} ProjectEmployee;

int compare(const void* a, const void* b) {
    ProjectEmployee* pa = (ProjectEmployee*)a;
    ProjectEmployee* pb = (ProjectEmployee*)b;
    
    if (pa->employee_id != pb->employee_id) {
        return pa->employee_id - pb->employee_id;
    }
    return pa->project_id - pb->project_id;
}

ProjectEmployee* solution(int projects[][3], int project_count,
                         char employees[][3][50], int employee_count,
                         int* result_count) {
    ProjectEmployee* joined = malloc(project_count * sizeof(ProjectEmployee));
    int joined_count = 0;
    
    // Join tables
    for (int i = 0; i < project_count; i++) {
        for (int j = 0; j < employee_count; j++) {
            if (projects[i][1] == atoi(employees[j][0])) {
                joined[joined_count].employee_id = projects[i][1];
                joined[joined_count].project_id = projects[i][0];
                joined[joined_count].workload = projects[i][2];
                strcpy(joined[joined_count].name, employees[j][1]);
                strcpy(joined[joined_count].team, employees[j][2]);
                joined_count++;
            }
        }
    }
    
    ProjectEmployee* result = malloc(joined_count * sizeof(ProjectEmployee));
    *result_count = 0;
    
    for (int i = 0; i < joined_count; i++) {
        // Calculate team average
        int team_workload_sum = 0;
        int team_member_count = 0;
        
        for (int j = 0; j < joined_count; j++) {
            if (strcmp(joined[j].team, joined[i].team) == 0) {
                team_workload_sum += joined[j].workload;
                team_member_count++;
            }
        }
        
        double team_avg = (double)team_workload_sum / team_member_count;
        
        if (joined[i].workload > team_avg) {
            result[(*result_count)++] = joined[i];
        }
    }
    
    // Sort results
    qsort(result, *result_count, sizeof(ProjectEmployee), compare);
    
    free(joined);
    return result;
}