day9: part2

master
Avril 4 years ago
parent 7300a4ebde
commit b07b60ad6f
Signed by: flanchan
GPG Key ID: 284488987C31F630

@ -3,6 +3,7 @@
#include <stdlib.h> #include <stdlib.h>
#include <string.h> #include <string.h>
#include <stdio.h> #include <stdio.h>
#include <assert.h>
#include "map.h" #include "map.h"
@ -52,47 +53,38 @@ static u64 csum(size_t i, size_t j)
sum += input[i]; sum += input[i];
return sum; return sum;
} }
static void reduce_bound(size_t* restrict i, size_t* restrict j, bool ex_by_high) static int _comp_u64(const void* _i, const void* _j)
{ {
register u64 ibound = 0; int iset=0; const u64* i = _i;
register u64 jbound = 0; int jset=0; const u64* j = _j;
if(*i<input_sz) (iset = 1, ibound = input[*i+1]); return *i < *j ? -1
if(*j!=0) (jset = -1, jbound = input[*j-1]); : *i > *j ? 1
: 0;
if(ex_by_high)
*(ibound>jbound ? i : j) += (ibound>jbound ? iset : jset);
else
*(ibound<jbound ? i : j) += (ibound<jbound ? iset: jset);
} }
static void extend_bound(size_t* restrict i, size_t* restrict j, bool ex_by_high) inline static u64 ud_sort(size_t i, size_t j)
{ {
register u64 ibound = 0; int iset=0; u64 slice[input_sz];
register u64 jbound = 0; int jset=0; size_t len = j-i;
assert(len<input_sz);
memcpy(slice, input+i, sizeof(u64)* len);
if(*i!=0) (iset = -1, ibound = input[*i-1]); qsort(slice, len, sizeof(u64), &_comp_u64);
if(*j<input_sz) (jset = 1, jbound = input[*j+1]);
if(ex_by_high) return slice[0] + slice[len-1];
*(ibound>jbound ? i : j) += (ibound>jbound ? iset : jset);
else
*(ibound<jbound ? i : j) += (ibound<jbound ? iset: jset);
} }
static u64 find_set(u64 target) static u64 find_set(u64 target)
{ {
size_t i=0,j=0; for(register size_t i =0;i<input_sz;i++)
while(1)
{ {
u64 sum = csum(i,j); for(register size_t j=i;j<input_sz;j++)
{
u64 sum = csum(i, j);
if(sum == target) { if(sum == target) {
printf("%lu\n", input[i] + input[j]); printf("%lu\n", ud_sort(i, j));
return 0; return 0;
} } else if (sum > target) break;
else if(sum < target) {
extend_bound(&i, &j, true);
} else if(sum > target) {
reduce_bound(&i, &j, true);
} }
} }
fprintf(stderr, "No set found\n"); fprintf(stderr, "No set found\n");

Loading…
Cancel
Save