Untitled
unknown
c_cpp
3 years ago
5.1 kB
7
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...