Untitled
unknown
c_cpp
2 years ago
5.1 kB
3
Indexable
#include <cs50.h> #include <stdio.h> #include <string.h> #include <ctype.h> #include <math.h> #include <stdlib.h> // Max number of candidates #define MAX 9 // preferences[i][j] is number of voters who prefer i over j int preferences[MAX][MAX]; // locked[i][j] means i is locked in over j bool locked[MAX][MAX]; // Each pair has a winner, loser typedef struct { int winner; int loser; } pair; // Array of candidates string candidates[MAX]; pair pairs[MAX * (MAX - 1) / 2]; int pair_count; int candidate_count; // Function prototypes bool vote(int rank, string name, int ranks[]); bool check(int candidate, int pair_number, int constant_candidate); void record_preferences(int ranks[]); void add_pairs(void); void sort_pairs(void); void lock_pairs(void); void print_winner(void); int main(int argc, string argv[]) { // Check for invalid usage if (argc < 2) { printf("Usage: tideman [candidate ...]\n"); return 1; } // Populate array of candidates candidate_count = argc - 1; if (candidate_count > MAX) { printf("Maximum number of candidates is %i\n", MAX); return 2; } for (int i = 0; i < candidate_count; i++) { candidates[i] = argv[i + 1]; } // Clear graph of locked in pairs for (int i = 0; i < candidate_count; i++) { for (int j = 0; j < candidate_count; j++) { locked[i][j] = false; } } pair_count = 0; int voter_count = get_int("Number of voters: "); // Query for votes for (int i = 0; i < voter_count; i++) { // ranks[i] is voter's ith preference int ranks[candidate_count]; // Query for each rank for (int j = 0; j < candidate_count; j++) { string name = get_string("Rank %i: ", j + 1); if (!vote(j, name, ranks)) { printf("Invalid vote.\n"); return 3; } } record_preferences(ranks); printf("\n"); } add_pairs(); sort_pairs(); lock_pairs(); print_winner(); return 0; } // Update ranks given a new vote bool vote(int rank, string name, int ranks[]) { for (int i = 0; i < candidate_count; i++) { if (strcmp(candidates[i], name) == 0) { ranks[rank] = i; return true; } } return false; } // Update preferences given one voter's ranks void record_preferences(int ranks[]) { for (int i = 0; i < candidate_count; i++) { for (int j = i + 1; j < candidate_count; j++) { preferences[ranks[i]][ranks[j]]++; } } for (int i = 0; i < candidate_count; i++) { preferences[ranks[i]][ranks[i]] = 0; } return; } // Record pairs of candidates where one is preferred over the other void add_pairs(void) { int k = 0; for (int i = 0; i < candidate_count; i++) { for (int j = 0; j < candidate_count; j++) { if (preferences[i][j] > preferences[j][i]) { pairs[k].winner = i; pairs[k].loser = j; pair_count++; k++; } } } return; } // Sort pairs in decreasing order by strength of victory void sort_pairs(void) { for (int i = 0; i < pair_count - 1; i++) { for (int j = 0; j < pair_count - 1; j++) { if (preferences[pairs[j].winner][pairs[j].loser] < preferences[pairs[j + 1].winner][pairs[j + 1].loser]) { pair cache = pairs[j]; pairs[j] = pairs[j + 1]; pairs[j + 1] = cache; } } } return; } // Lock pairs into the candidate graph in order, without creating cycles void lock_pairs(void) { for (int i = 0; i < pair_count; i++) { if (check(pairs[i].loser, i, pairs[i].winner) == true) { locked[pairs[i].winner][pairs[i].loser] = false; } else { locked[pairs[i].winner][pairs[i].loser] = true; } } return; } bool check(int candidate, int pair_number, int constant_candidate) { if (candidate == constant_candidate) { return true; } for (int i = 0; i < pair_number; i++) { if (candidate == pairs[i].winner) { check(pairs[i].loser, pair_number, constant_candidate); if (check(pairs[i].loser, pair_number, constant_candidate) == true) { return true; } } } return false; } // Print the winner of the election void print_winner(void) { int check; for (int k = 0, j = 0; k < candidate_count; k++, j++) { check = 0; for (int i = 0; i < candidate_count; i++) { if (locked[i][j] == false) { check++; } if (check == candidate_count) { printf("%s\n", candidates[j]); } } } return; }
Editor is loading...