100 lines
2.3 KiB
C
100 lines
2.3 KiB
C
/* aoc-c.c: Advent of Code 2022, day 1
|
|
*
|
|
* Copyright (C) 2022 Bruno Raoult ("br")
|
|
* Licensed under the GNU General Public License v3.0 or later.
|
|
* Some rights reserved. See COPYING.
|
|
*
|
|
* You should have received a copy of the GNU General Public License along with this
|
|
* program. If not, see <https://www.gnu.org/licenses/gpl-3.0-standalone.html>.
|
|
*
|
|
* SPDX-License-Identifier: GPL-3.0-or-later <https://spdx.org/licenses/GPL-3.0-or-later.html>
|
|
*/
|
|
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <unistd.h>
|
|
|
|
#include "plist.h"
|
|
#include "debug.h"
|
|
#include "pool.h"
|
|
|
|
PLIST_HEAD(plist);
|
|
|
|
static int calc_top_plist(int n)
|
|
{
|
|
int res = 0;
|
|
|
|
struct plist_node *node;
|
|
plist_for_each_reverse(node, &plist) {
|
|
res += node->prio;
|
|
if (!--n)
|
|
break;
|
|
}
|
|
return res;
|
|
}
|
|
|
|
static void parse(pool_t *pool)
|
|
{
|
|
size_t alloc = 0;
|
|
char *buf = NULL;
|
|
ssize_t buflen;
|
|
int total = 0;
|
|
struct plist_node *node;
|
|
|
|
while (1) {
|
|
buflen = getline(&buf, &alloc, stdin);
|
|
switch (buflen) {
|
|
case 1:
|
|
case -1:
|
|
node = pool_get(pool);
|
|
plist_node_init(node, total);
|
|
plist_add(node, &plist);
|
|
total = 0;
|
|
if (buflen == -1)
|
|
goto end;
|
|
break;
|
|
default:
|
|
total += atoi(buf);
|
|
}
|
|
}
|
|
end:
|
|
free(buf);
|
|
return;
|
|
}
|
|
|
|
static int usage(char *prg)
|
|
{
|
|
fprintf(stderr, "Usage: %s [-d debug_level] [-p part] [-i input]\n", prg);
|
|
return 1;
|
|
}
|
|
|
|
int main(int ac, char **av)
|
|
{
|
|
int opt, part = 1;
|
|
|
|
while ((opt = getopt(ac, av, "d:p:")) != -1) {
|
|
switch (opt) {
|
|
case 'd':
|
|
debug_level_set(atoi(optarg));
|
|
break;
|
|
case 'p': /* 1 or 2 */
|
|
part = atoi(optarg);
|
|
if (part < 1 || part > 2)
|
|
return usage(*av);
|
|
break;
|
|
default:
|
|
return usage(*av);
|
|
}
|
|
}
|
|
if (optind < ac)
|
|
return usage(*av);
|
|
|
|
pool_t *pool_tot = pool_create("total", 128, sizeof(struct plist_node));
|
|
|
|
parse(pool_tot);
|
|
|
|
printf("%s : res=%d\n", *av, calc_top_plist(part == 1? 1: 3));
|
|
pool_destroy(pool_tot);
|
|
exit(0);
|
|
}
|