← All problems Problem 871
Drifting Subsets — max drifting size sum for f_n(x)=x^3+x+1 mod n, n=100001..100100.
View problem on Project Euler
Performance comparison
Metric Our solution Best known
Time complexity O(n^2)O(n * sum)Space complexity O(n^2)O(sum)Approach Flow solution Subset sum DP Verdict Unknown
Flow source
# Project Euler 871
# Drifting Subsets — max drifting size sum for f_n(x)=x^3+x+1 mod n, n=100001..100100.
extern {
function calloc(n: i64, size: i64) -> ptr<void>
function free(p: ptr<void>) -> void
}
function drifting_subset_size(n: i64) -> i64 {
let mut f: ptr<i64> = calloc(n, 8)
let mut indeg: ptr<i64> = calloc(n, 8)
let mut x: i64 = 0
while x < n {
let y: i64 = ((x * x % n) * x + x + 1) % n
f[x] = y
indeg[y] = indeg[y] + 1
x = x + 1
}
let mut sum_a: ptr<i64> = calloc(n, 8)
let mut best: ptr<i8> = calloc(n, 1)
let mut q: ptr<i64> = calloc(n, 8)
let mut qh: i64 = 0
let mut qt: i64 = 0
x = 0
while x < n {
if indeg[x] == 0 {
q[qt] = x
qt = qt + 1
}
x = x + 1
}
while qh < qt {
let u: i64 = q[qh]
qh = qh + 1
let dp0_u: i64 = sum_a[u] + (best[u] as i64)
let v: i64 = f[u]
sum_a[v] = sum_a[v] + dp0_u
if best[u] == 0 { best[v] = 1 }
indeg[v] = indeg[v] - 1
if indeg[v] == 0 {
q[qt] = v
qt = qt + 1
}
}
let mut visited: ptr<i8> = calloc(n, 1)
let mut total: i64 = 0
let mut start: i64 = 0
while start < n {
if indeg[start] != 0 && visited[start] == 0 {
let mut cycle: ptr<i64> = calloc(n, 8)
let mut k: i64 = 0
let mut cur: i64 = start
while visited[cur] == 0 {
visited[cur] = 1
cycle[k] = cur
k = k + 1
cur = f[cur]
}
if k == 1 {
let v: i64 = cycle[0]
total = total + sum_a[v] + (best[v] as i64)
} else {
let mut deltas: ptr<i64> = calloc(k, 8)
let mut base: i64 = 0
let mut i: i64 = 0
while i < k {
let v: i64 = cycle[i]
let d: i64 = best[v] as i64
deltas[i] = d
base = base + sum_a[v] + d
i = i + 1
}
let mut dp2: i64 = 0
let mut dp1: i64 = 0
i = 0
while i < k - 1 {
let w: i64 = 1 - deltas[i] - deltas[i + 1]
let mut take: i64 = dp2 + w
let mut dp: i64 = dp1
if take > dp1 { dp = take }
dp2 = dp1
dp1 = dp
i = i + 1
}
let gain_a: i64 = dp1
let w_last: i64 = 1 - deltas[k - 1] - deltas[0]
let mut gain_b: i64 = w_last
if k > 2 {
dp2 = 0
dp1 = 0
i = 1
while i < k - 2 {
let w: i64 = 1 - deltas[i] - deltas[i + 1]
let mut take: i64 = dp2 + w
let mut dp: i64 = dp1
if take > dp1 { dp = take }
dp2 = dp1
dp1 = dp
i = i + 1
}
gain_b = w_last + dp1
}
let mut best_cycle_gain: i64 = gain_a
if gain_b > best_cycle_gain { best_cycle_gain = gain_b }
if best_cycle_gain < 0 { best_cycle_gain = 0 }
total = total + base + best_cycle_gain
free(deltas)
}
free(cycle)
}
start = start + 1
}
free(f)
free(indeg)
free(sum_a)
free(best)
free(q)
free(visited)
return total
}
function main() -> i32 {
let mut total: i64 = 0
let mut i: i64 = 1
while i <= 100 {
total = total + drifting_subset_size(100000 + i)
i = i + 1
}
printf("%lld\n", total)
return 0
}
Generated C
#include <stdint.h>
#include <stdbool.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
/* Flow runtime helpers */
typedef struct flow_temp_node { struct flow_temp_node* next; } flow_temp_node;
static flow_temp_node* flow_temp_head = NULL;
static int flow_temp_atexit_set = 0;
__attribute__((unused)) static void flow_temp_free_all(void) {
while (flow_temp_head) {
flow_temp_node* n = flow_temp_head;
flow_temp_head = n->next;
free(n);
}
}
__attribute__((unused)) static void* flow_temp_alloc(size_t nbytes) {
flow_temp_node* node = (flow_temp_node*)malloc(sizeof(flow_temp_node) + nbytes);
if (!node) return NULL;
node->next = flow_temp_head;
flow_temp_head = node;
if (!flow_temp_atexit_set) {
flow_temp_atexit_set = 1;
atexit(flow_temp_free_all);
}
return (void*)(node + 1);
}
#ifndef FLOW_DIAG
#define FLOW_DIAG(msg) fprintf(stderr, "%s", (msg))
#endif
#ifndef FLOW_LOG
#define FLOW_LOG(fmt, ...) printf(fmt, __VA_ARGS__)
#endif
#ifndef FLOW_LOG_EMPTY
#define FLOW_LOG_EMPTY(fmt) printf(fmt)
#endif
static char* flow_strcat(const char* a, const char* b) {
size_t la = strlen(a ? a : ""), lb = strlen(b ? b : "");
char* r = (char*)flow_temp_alloc(la + lb + 1);
if (!r) return NULL;
if (la) memcpy(r, a, la);
if (lb) memcpy(r + la, b, lb);
r[la + lb] = '\0';
return r;
}
#define __flow_in_arr(arr, val) __extension__ ({ \
int _found = 0; \
size_t _n = sizeof(arr)/sizeof((arr)[0]); \
for (size_t _i = 0; _i < _n; _i++) { \
if ((arr)[_i] == (val)) { _found = 1; break; } \
} _found; })
/* Unified fault handler (MISRA #279) — override with -DFLOW_FAULT_HANDLER=fn */
#ifndef FLOW_FAULT_HANDLER
__attribute__((unused)) static inline void flow_fault_handler(const char* msg) {
fprintf(stderr, "flow: %s\n", msg ? msg : "fault");
abort();
#if defined(__GNUC__) || defined(__clang__)
__builtin_unreachable();
#endif
}
#else
#define flow_fault_handler FLOW_FAULT_HANDLER
#endif
#define flow_div_by_zero_handler() flow_fault_handler("division by zero")
#define flow_shift_ub_handler() flow_fault_handler("invalid shift (amount out of range or left-shift of negative)")
#ifndef FLOW_CHECKED_DIV
#define FLOW_CHECKED_DIV(L, R) (((R) != 0) ? ((L) / (R)) : (flow_div_by_zero_handler(), (L) * 0))
#endif
#ifndef FLOW_CHECKED_MOD
#define FLOW_CHECKED_MOD(L, R) (((R) != 0) ? ((L) % (R)) : (flow_div_by_zero_handler(), (L) * 0))
#endif
#ifndef FLOW_CHECKED_SHL
#define FLOW_CHECKED_SHL(L, R) ((((R) >= 0) && ((unsigned long long)(R) < (sizeof(L) * 8ull)) && ((L) >= 0)) ? ((L) << (R)) : (flow_shift_ub_handler(), (L) * 0))
#endif
#ifndef FLOW_CHECKED_SHR
#define FLOW_CHECKED_SHR(L, R) ((((R) >= 0) && ((unsigned long long)(R) < (sizeof(L) * 8ull))) ? ((L) >> (R)) : (flow_shift_ub_handler(), (L) * 0))
#endif
#include <math.h>
void* _ui_state = NULL;
static inline float i32_to_f32(int32_t v) { return (float)v; }
/* Host stub for @gpu kernels (device codegen replaces this). */
static inline int32_t gpu_thread_id(void) { return 0; }
int64_t drifting_subset_size_i64(int64_t n);
int32_t main(void);
int64_t drifting_subset_size_i64(int64_t n) {
int64_t* f = (int64_t*)(calloc(n, 8));
int64_t* indeg = (int64_t*)(calloc(n, 8));
int64_t x = 0;
while (x < n) {
int64_t y = FLOW_CHECKED_MOD(((((FLOW_CHECKED_MOD(((x * x)), (n)) * x) + x) + 1)), (n));
f[x] = y;
indeg[y] = (indeg[y] + 1);
x = (x + 1);
}
int64_t* sum_a = (int64_t*)(calloc(n, 8));
int8_t* best = (int8_t*)(calloc(n, 1));
int64_t* q = (int64_t*)(calloc(n, 8));
int64_t qh = 0;
int64_t qt = 0;
x = 0;
while (x < n) {
if (indeg[x] == 0) {
q[qt] = x;
qt = (qt + 1);
}
x = (x + 1);
}
while (qh < qt) {
int64_t u = q[qh];
qh = (qh + 1);
int64_t dp0_u = (sum_a[u] + ((int64_t)(best[u])));
int64_t v = f[u];
sum_a[v] = (sum_a[v] + dp0_u);
if (best[u] == 0) {
best[v] = 1;
}
indeg[v] = (indeg[v] - 1);
if (indeg[v] == 0) {
q[qt] = v;
qt = (qt + 1);
}
}
int8_t* visited = (int8_t*)(calloc(n, 1));
int64_t total = 0;
int64_t start = 0;
while (start < n) {
if ((indeg[start] != 0 && visited[start] == 0)) {
int64_t* cycle = (int64_t*)(calloc(n, 8));
int64_t k = 0;
int64_t cur = start;
while (visited[cur] == 0) {
visited[cur] = 1;
cycle[k] = cur;
k = (k + 1);
cur = f[cur];
}
if (k == 1) {
int64_t v = cycle[0];
total = ((total + sum_a[v]) + ((int64_t)(best[v])));
} else {
int64_t* deltas = (int64_t*)(calloc(k, 8));
int64_t base = 0;
int64_t i = 0;
while (i < k) {
int64_t v = cycle[i];
int64_t d = ((int64_t)(best[v]));
deltas[i] = d;
base = ((base + sum_a[v]) + d);
i = (i + 1);
}
int64_t dp2 = 0;
int64_t dp1 = 0;
i = 0;
while (i < (k - 1)) {
int64_t w = ((1 - deltas[i]) - deltas[(i + 1)]);
int64_t take = (dp2 + w);
int64_t dp = dp1;
if (take > dp1) {
dp = take;
}
dp2 = dp1;
dp1 = dp;
i = (i + 1);
}
int64_t gain_a = dp1;
int64_t w_last = ((1 - deltas[(k - 1)]) - deltas[0]);
int64_t gain_b = w_last;
if (k > 2) {
dp2 = 0;
dp1 = 0;
i = 1;
while (i < (k - 2)) {
int64_t w = ((1 - deltas[i]) - deltas[(i + 1)]);
int64_t take = (dp2 + w);
int64_t dp = dp1;
if (take > dp1) {
dp = take;
}
dp2 = dp1;
dp1 = dp;
i = (i + 1);
}
gain_b = (w_last + dp1);
}
int64_t best_cycle_gain = gain_a;
if (gain_b > best_cycle_gain) {
best_cycle_gain = gain_b;
}
if (best_cycle_gain < 0) {
best_cycle_gain = 0;
}
total = ((total + base) + best_cycle_gain);
free(deltas);
}
free(cycle);
}
start = (start + 1);
}
free(f);
free(indeg);
free(sum_a);
free(best);
free(q);
free(visited);
return total;
}
int32_t main(void) {
int64_t total = 0;
int64_t i = 1;
while (i <= 100) {
total = (total + drifting_subset_size_i64((100000 + i)));
i = (i + 1);
}
printf("%lld\n", total);
return 0;
}
Generated MLIR
module {
llvm.func @printf(!llvm.ptr, ...) -> i32
llvm.mlir.global internal constant @str_0("%lld\n\00") {addr_space = 0 : i32} : !llvm.array<6 x i8>
func.func private @calloc(i64, i64) -> !llvm.ptr
func.func private @free(!llvm.ptr) -> ()
func.func @drifting_subset_size(%arg0: i64) -> i64 {
%1 = arith.constant 8 : i32
%2 = arith.extsi %1 : i32 to i64
%0 = func.call @calloc(%arg0, %2) : (i64, i64) -> !llvm.ptr
%3 = llvm.mlir.constant(1 : i64) : i64
%4 = llvm.alloca %3 x !llvm.ptr : (i64) -> !llvm.ptr
llvm.store %0, %4 : !llvm.ptr, !llvm.ptr
%6 = arith.constant 8 : i32
%7 = arith.extsi %6 : i32 to i64
%5 = func.call @calloc(%arg0, %7) : (i64, i64) -> !llvm.ptr
%8 = llvm.mlir.constant(1 : i64) : i64
%9 = llvm.alloca %8 x !llvm.ptr : (i64) -> !llvm.ptr
llvm.store %5, %9 : !llvm.ptr, !llvm.ptr
%10 = arith.constant 0 : i32
%11 = arith.extsi %10 : i32 to i64
%12 = llvm.mlir.constant(1 : i64) : i64
%13 = llvm.alloca %12 x i64 : (i64) -> !llvm.ptr
llvm.store %11, %13 : i64, !llvm.ptr
cf.br ^bb0
^bb0:
%14 = llvm.load %13 : !llvm.ptr -> i64
%15 = arith.cmpi slt, %14, %arg0 : i64
cf.cond_br %15, ^bb1, ^bb2
^bb1:
%16 = llvm.load %13 : !llvm.ptr -> i64
%17 = llvm.load %13 : !llvm.ptr -> i64
%18 = arith.muli %16, %17 : i64
%19 = arith.remsi %18, %arg0 : i64
%20 = llvm.load %13 : !llvm.ptr -> i64
%21 = arith.muli %19, %20 : i64
%22 = llvm.load %13 : !llvm.ptr -> i64
%23 = arith.addi %21, %22 : i64
%24 = arith.constant 1 : i32
%26 = arith.extsi %24 : i32 to i64
%25 = arith.addi %23, %26 : i64
%27 = arith.remsi %25, %arg0 : i64
%28 = llvm.load %4 : !llvm.ptr -> !llvm.ptr
%29 = llvm.load %13 : !llvm.ptr -> i64
%30 = llvm.getelementptr %28[%29] : (!llvm.ptr, i64) -> !llvm.ptr, i64
llvm.store %27, %30 : i64, !llvm.ptr
%32 = llvm.load %9 : !llvm.ptr -> !llvm.ptr
%33 = llvm.getelementptr %32[%27] : (!llvm.ptr, i64) -> !llvm.ptr, i64
%31 = llvm.load %33 : !llvm.ptr -> i64
%34 = arith.constant 1 : i32
%36 = arith.extsi %34 : i32 to i64
%35 = arith.addi %31, %36 : i64
%37 = llvm.load %9 : !llvm.ptr -> !llvm.ptr
%38 = llvm.getelementptr %37[%27] : (!llvm.ptr, i64) -> !llvm.ptr, i64
llvm.store %35, %38 : i64, !llvm.ptr
%39 = llvm.load %13 : !llvm.ptr -> i64
%40 = arith.constant 1 : i32
%42 = arith.extsi %40 : i32 to i64
%41 = arith.addi %39, %42 : i64
llvm.store %41, %13 : i64, !llvm.ptr
cf.br ^bb0
^bb2:
%44 = arith.constant 8 : i32
%45 = arith.extsi %44 : i32 to i64
%43 = func.call @calloc(%arg0, %45) : (i64, i64) -> !llvm.ptr
%46 = llvm.mlir.constant(1 : i64) : i64
%47 = llvm.alloca %46 x !llvm.ptr : (i64) -> !llvm.ptr
llvm.store %43, %47 : !llvm.ptr, !llvm.ptr
%49 = arith.constant 1 : i32
%50 = arith.extsi %49 : i32 to i64
%48 = func.call @calloc(%arg0, %50) : (i64, i64) -> !llvm.ptr
%51 = llvm.mlir.constant(1 : i64) : i64
%52 = llvm.alloca %51 x !llvm.ptr : (i64) -> !llvm.ptr
llvm.store %48, %52 : !llvm.ptr, !llvm.ptr
%54 = arith.constant 8 : i32
%55 = arith.extsi %54 : i32 to i64
%53 = func.call @calloc(%arg0, %55) : (i64, i64) -> !llvm.ptr
%56 = llvm.mlir.constant(1 : i64) : i64
%57 = llvm.alloca %56 x !llvm.ptr : (i64) -> !llvm.ptr
llvm.store %53, %57 : !llvm.ptr, !llvm.ptr
%58 = arith.constant 0 : i32
%59 = arith.extsi %58 : i32 to i64
%60 = llvm.mlir.constant(1 : i64) : i64
%61 = llvm.alloca %60 x i64 : (i64) -> !llvm.ptr
llvm.store %59, %61 : i64, !llvm.ptr
%62 = arith.constant 0 : i32
%63 = arith.extsi %62 : i32 to i64
%64 = llvm.mlir.constant(1 : i64) : i64
%65 = llvm.alloca %64 x i64 : (i64) -> !llvm.ptr
llvm.store %63, %65 : i64, !llvm.ptr
%66 = arith.constant 0 : i32
%67 = arith.extsi %66 : i32 to i64
llvm.store %67, %13 : i64, !llvm.ptr
cf.br ^bb3
^bb3:
%68 = llvm.load %13 : !llvm.ptr -> i64
%69 = arith.cmpi slt, %68, %arg0 : i64
cf.cond_br %69, ^bb4, ^bb5
^bb4:
%71 = llvm.load %9 : !llvm.ptr -> !llvm.ptr
%72 = llvm.load %13 : !llvm.ptr -> i64
%73 = llvm.getelementptr %71[%72] : (!llvm.ptr, i64) -> !llvm.ptr, i64
%70 = llvm.load %73 : !llvm.ptr -> i64
%74 = arith.constant 0 : i32
%76 = arith.extsi %74 : i32 to i64
%75 = arith.cmpi eq, %70, %76 : i64
cf.cond_br %75, ^bb6, ^bb7
^bb6:
%77 = llvm.load %13 : !llvm.ptr -> i64
%78 = llvm.load %57 : !llvm.ptr -> !llvm.ptr
%79 = llvm.load %65 : !llvm.ptr -> i64
%80 = llvm.getelementptr %78[%79] : (!llvm.ptr, i64) -> !llvm.ptr, i64
llvm.store %77, %80 : i64, !llvm.ptr
%81 = llvm.load %65 : !llvm.ptr -> i64
%82 = arith.constant 1 : i32
%84 = arith.extsi %82 : i32 to i64
%83 = arith.addi %81, %84 : i64
llvm.store %83, %65 : i64, !llvm.ptr
cf.br ^bb8
^bb7:
cf.br ^bb8
^bb8:
%85 = llvm.load %13 : !llvm.ptr -> i64
%86 = arith.constant 1 : i32
%88 = arith.extsi %86 : i32 to i64
%87 = arith.addi %85, %88 : i64
llvm.store %87, %13 : i64, !llvm.ptr
cf.br ^bb3
^bb5:
cf.br ^bb9
^bb9:
%89 = llvm.load %61 : !llvm.ptr -> i64
%90 = llvm.load %65 : !llvm.ptr -> i64
%91 = arith.cmpi slt, %89, %90 : i64
cf.cond_br %91, ^bb10, ^bb11
^bb10:
%93 = llvm.load %57 : !llvm.ptr -> !llvm.ptr
%94 = llvm.load %61 : !llvm.ptr -> i64
%95 = llvm.getelementptr %93[%94] : (!llvm.ptr, i64) -> !llvm.ptr, i64
%92 = llvm.load %95 : !llvm.ptr -> i64
%96 = llvm.load %61 : !llvm.ptr -> i64
%97 = arith.constant 1 : i32
%99 = arith.extsi %97 : i32 to i64
%98 = arith.addi %96, %99 : i64
llvm.store %98, %61 : i64, !llvm.ptr
%101 = llvm.load %47 : !llvm.ptr -> !llvm.ptr
%102 = llvm.getelementptr %101[%92] : (!llvm.ptr, i64) -> !llvm.ptr, i64
%100 = llvm.load %102 : !llvm.ptr -> i64
%104 = llvm.load %52 : !llvm.ptr -> !llvm.ptr
%105 = llvm.getelementptr %104[%92] : (!llvm.ptr, i64) -> !llvm.ptr, i8
%103 = llvm.load %105 : !llvm.ptr -> i8
%106 = arith.extsi %103 : i8 to i64
%107 = arith.addi %100, %106 : i64
%109 = llvm.load %4 : !llvm.ptr -> !llvm.ptr
%110 = llvm.getelementptr %109[%92] : (!llvm.ptr, i64) -> !llvm.ptr, i64
%108 = llvm.load %110 : !llvm.ptr -> i64
%112 = llvm.load %47 : !llvm.ptr -> !llvm.ptr
%113 = llvm.getelementptr %112[%108] : (!llvm.ptr, i64) -> !llvm.ptr, i64
%111 = llvm.load %113 : !llvm.ptr -> i64
%114 = arith.addi %111, %107 : i64
%115 = llvm.load %47 : !llvm.ptr -> !llvm.ptr
%116 = llvm.getelementptr %115[%108] : (!llvm.ptr, i64) -> !llvm.ptr, i64
llvm.store %114, %116 : i64, !llvm.ptr
%118 = llvm.load %52 : !llvm.ptr -> !llvm.ptr
%119 = llvm.getelementptr %118[%92] : (!llvm.ptr, i64) -> !llvm.ptr, i8
%117 = llvm.load %119 : !llvm.ptr -> i8
%120 = arith.constant 0 : i32
%122 = arith.extsi %117 : i8 to i32
%121 = arith.cmpi eq, %122, %120 : i32
cf.cond_br %121, ^bb12, ^bb13
^bb12:
%123 = arith.constant 1 : i32
%124 = llvm.load %52 : !llvm.ptr -> !llvm.ptr
%125 = arith.trunci %123 : i32 to i8
%126 = llvm.getelementptr %124[%108] : (!llvm.ptr, i64) -> !llvm.ptr, i8
llvm.store %125, %126 : i8, !llvm.ptr
cf.br ^bb14
^bb13:
cf.br ^bb14
^bb14:
%128 = llvm.load %9 : !llvm.ptr -> !llvm.ptr
%129 = llvm.getelementptr %128[%108] : (!llvm.ptr, i64) -> !llvm.ptr, i64
%127 = llvm.load %129 : !llvm.ptr -> i64
%130 = arith.constant 1 : i32
%132 = arith.extsi %130 : i32 to i64
%131 = arith.subi %127, %132 : i64
%133 = llvm.load %9 : !llvm.ptr -> !llvm.ptr
%134 = llvm.getelementptr %133[%108] : (!llvm.ptr, i64) -> !llvm.ptr, i64
llvm.store %131, %134 : i64, !llvm.ptr
%136 = llvm.load %9 : !llvm.ptr -> !llvm.ptr
%137 = llvm.getelementptr %136[%108] : (!llvm.ptr, i64) -> !llvm.ptr, i64
%135 = llvm.load %137 : !llvm.ptr -> i64
%138 = arith.constant 0 : i32
%140 = arith.extsi %138 : i32 to i64
%139 = arith.cmpi eq, %135, %140 : i64
cf.cond_br %139, ^bb15, ^bb16
^bb15:
%141 = llvm.load %57 : !llvm.ptr -> !llvm.ptr
%142 = llvm.load %65 : !llvm.ptr -> i64
%143 = llvm.getelementptr %141[%142] : (!llvm.ptr, i64) -> !llvm.ptr, i64
llvm.store %108, %143 : i64, !llvm.ptr
%144 = llvm.load %65 : !llvm.ptr -> i64
%145 = arith.constant 1 : i32
%147 = arith.extsi %145 : i32 to i64
%146 = arith.addi %144, %147 : i64
llvm.store %146, %65 : i64, !llvm.ptr
cf.br ^bb17
^bb16:
cf.br ^bb17
^bb17:
cf.br ^bb9
^bb11:
%149 = arith.constant 1 : i32
%150 = arith.extsi %149 : i32 to i64
%148 = func.call @calloc(%arg0, %150) : (i64, i64) -> !llvm.ptr
%151 = llvm.mlir.constant(1 : i64) : i64
%152 = llvm.alloca %151 x !llvm.ptr : (i64) -> !llvm.ptr
llvm.store %148, %152 : !llvm.ptr, !llvm.ptr
%153 = arith.constant 0 : i32
%154 = arith.extsi %153 : i32 to i64
%155 = llvm.mlir.constant(1 : i64) : i64
%156 = llvm.alloca %155 x i64 : (i64) -> !llvm.ptr
llvm.store %154, %156 : i64, !llvm.ptr
%157 = arith.constant 0 : i32
%158 = arith.extsi %157 : i32 to i64
%159 = llvm.mlir.constant(1 : i64) : i64
%160 = llvm.alloca %159 x i64 : (i64) -> !llvm.ptr
llvm.store %158, %160 : i64, !llvm.ptr
cf.br ^bb18
^bb18:
%161 = llvm.load %160 : !llvm.ptr -> i64
%162 = arith.cmpi slt, %161, %arg0 : i64
cf.cond_br %162, ^bb19, ^bb20
^bb19:
%164 = llvm.load %9 : !llvm.ptr -> !llvm.ptr
%165 = llvm.load %160 : !llvm.ptr -> i64
%166 = llvm.getelementptr %164[%165] : (!llvm.ptr, i64) -> !llvm.ptr, i64
%163 = llvm.load %166 : !llvm.ptr -> i64
%167 = arith.constant 0 : i32
%169 = arith.extsi %167 : i32 to i64
%168 = arith.cmpi ne, %163, %169 : i64
%170 = scf.if %168 -> (i1) {
%172 = llvm.load %152 : !llvm.ptr -> !llvm.ptr
%173 = llvm.load %160 : !llvm.ptr -> i64
%174 = llvm.getelementptr %172[%173] : (!llvm.ptr, i64) -> !llvm.ptr, i8
%171 = llvm.load %174 : !llvm.ptr -> i8
%175 = arith.constant 0 : i32
%177 = arith.extsi %171 : i8 to i32
%176 = arith.cmpi eq, %177, %175 : i32
scf.yield %176 : i1
} else {
%178 = arith.constant false
scf.yield %178 : i1
}
cf.cond_br %170, ^bb21, ^bb22
^bb21:
%180 = arith.constant 8 : i32
%181 = arith.extsi %180 : i32 to i64
%179 = func.call @calloc(%arg0, %181) : (i64, i64) -> !llvm.ptr
%182 = llvm.mlir.constant(1 : i64) : i64
%183 = llvm.alloca %182 x !llvm.ptr : (i64) -> !llvm.ptr
llvm.store %179, %183 : !llvm.ptr, !llvm.ptr
%184 = arith.constant 0 : i32
%185 = arith.extsi %184 : i32 to i64
%186 = llvm.mlir.constant(1 : i64) : i64
%187 = llvm.alloca %186 x i64 : (i64) -> !llvm.ptr
llvm.store %185, %187 : i64, !llvm.ptr
%188 = llvm.load %160 : !llvm.ptr -> i64
%189 = llvm.mlir.constant(1 : i64) : i64
%190 = llvm.alloca %189 x i64 : (i64) -> !llvm.ptr
llvm.store %188, %190 : i64, !llvm.ptr
cf.br ^bb24
^bb24:
%192 = llvm.load %152 : !llvm.ptr -> !llvm.ptr
%193 = llvm.load %190 : !llvm.ptr -> i64
%194 = llvm.getelementptr %192[%193] : (!llvm.ptr, i64) -> !llvm.ptr, i8
%191 = llvm.load %194 : !llvm.ptr -> i8
%195 = arith.constant 0 : i32
%197 = arith.extsi %191 : i8 to i32
%196 = arith.cmpi eq, %197, %195 : i32
cf.cond_br %196, ^bb25, ^bb26
^bb25:
%198 = arith.constant 1 : i32
%199 = llvm.load %152 : !llvm.ptr -> !llvm.ptr
%200 = llvm.load %190 : !llvm.ptr -> i64
%201 = arith.trunci %198 : i32 to i8
%202 = llvm.getelementptr %199[%200] : (!llvm.ptr, i64) -> !llvm.ptr, i8
llvm.store %201, %202 : i8, !llvm.ptr
%203 = llvm.load %190 : !llvm.ptr -> i64
%204 = llvm.load %183 : !llvm.ptr -> !llvm.ptr
%205 = llvm.load %187 : !llvm.ptr -> i64
%206 = llvm.getelementptr %204[%205] : (!llvm.ptr, i64) -> !llvm.ptr, i64
llvm.store %203, %206 : i64, !llvm.ptr
%207 = llvm.load %187 : !llvm.ptr -> i64
%208 = arith.constant 1 : i32
%210 = arith.extsi %208 : i32 to i64
%209 = arith.addi %207, %210 : i64
llvm.store %209, %187 : i64, !llvm.ptr
%212 = llvm.load %4 : !llvm.ptr -> !llvm.ptr
%213 = llvm.load %190 : !llvm.ptr -> i64
%214 = llvm.getelementptr %212[%213] : (!llvm.ptr, i64) -> !llvm.ptr, i64
%211 = llvm.load %214 : !llvm.ptr -> i64
llvm.store %211, %190 : i64, !llvm.ptr
cf.br ^bb24
^bb26:
%215 = llvm.load %187 : !llvm.ptr -> i64
%216 = arith.constant 1 : i32
%218 = arith.extsi %216 : i32 to i64
%217 = arith.cmpi eq, %215, %218 : i64
cf.cond_br %217, ^bb27, ^bb28
^bb27:
%220 = llvm.load %183 : !llvm.ptr -> !llvm.ptr
%221 = arith.constant 0 : i32
%222 = arith.extsi %221 : i32 to i64
%223 = llvm.getelementptr %220[%222] : (!llvm.ptr, i64) -> !llvm.ptr, i64
%219 = llvm.load %223 : !llvm.ptr -> i64
%224 = llvm.load %156 : !llvm.ptr -> i64
%226 = llvm.load %47 : !llvm.ptr -> !llvm.ptr
%227 = llvm.getelementptr %226[%219] : (!llvm.ptr, i64) -> !llvm.ptr, i64
%225 = llvm.load %227 : !llvm.ptr -> i64
%228 = arith.addi %224, %225 : i64
%230 = llvm.load %52 : !llvm.ptr -> !llvm.ptr
%231 = llvm.getelementptr %230[%219] : (!llvm.ptr, i64) -> !llvm.ptr, i8
%229 = llvm.load %231 : !llvm.ptr -> i8
%232 = arith.extsi %229 : i8 to i64
%233 = arith.addi %228, %232 : i64
llvm.store %233, %156 : i64, !llvm.ptr
cf.br ^bb29
^bb28:
%235 = llvm.load %187 : !llvm.ptr -> i64
%236 = arith.constant 8 : i32
%237 = arith.extsi %236 : i32 to i64
%234 = func.call @calloc(%235, %237) : (i64, i64) -> !llvm.ptr
%238 = llvm.mlir.constant(1 : i64) : i64
%239 = llvm.alloca %238 x !llvm.ptr : (i64) -> !llvm.ptr
llvm.store %234, %239 : !llvm.ptr, !llvm.ptr
%240 = arith.constant 0 : i32
%241 = arith.extsi %240 : i32 to i64
%242 = llvm.mlir.constant(1 : i64) : i64
%243 = llvm.alloca %242 x i64 : (i64) -> !llvm.ptr
llvm.store %241, %243 : i64, !llvm.ptr
%244 = arith.constant 0 : i32
%245 = arith.extsi %244 : i32 to i64
%246 = llvm.mlir.constant(1 : i64) : i64
%247 = llvm.alloca %246 x i64 : (i64) -> !llvm.ptr
llvm.store %245, %247 : i64, !llvm.ptr
cf.br ^bb30
^bb30:
%248 = llvm.load %247 : !llvm.ptr -> i64
%249 = llvm.load %187 : !llvm.ptr -> i64
%250 = arith.cmpi slt, %248, %249 : i64
cf.cond_br %250, ^bb31, ^bb32
^bb31:
%252 = llvm.load %183 : !llvm.ptr -> !llvm.ptr
%253 = llvm.load %247 : !llvm.ptr -> i64
%254 = llvm.getelementptr %252[%253] : (!llvm.ptr, i64) -> !llvm.ptr, i64
%251 = llvm.load %254 : !llvm.ptr -> i64
%256 = llvm.load %52 : !llvm.ptr -> !llvm.ptr
%257 = llvm.getelementptr %256[%251] : (!llvm.ptr, i64) -> !llvm.ptr, i8
%255 = llvm.load %257 : !llvm.ptr -> i8
%258 = arith.extsi %255 : i8 to i64
%259 = llvm.load %239 : !llvm.ptr -> !llvm.ptr
%260 = llvm.load %247 : !llvm.ptr -> i64
%261 = llvm.getelementptr %259[%260] : (!llvm.ptr, i64) -> !llvm.ptr, i64
llvm.store %258, %261 : i64, !llvm.ptr
%262 = llvm.load %243 : !llvm.ptr -> i64
%264 = llvm.load %47 : !llvm.ptr -> !llvm.ptr
%265 = llvm.getelementptr %264[%251] : (!llvm.ptr, i64) -> !llvm.ptr, i64
%263 = llvm.load %265 : !llvm.ptr -> i64
%266 = arith.addi %262, %263 : i64
%267 = arith.addi %266, %258 : i64
llvm.store %267, %243 : i64, !llvm.ptr
%268 = llvm.load %247 : !llvm.ptr -> i64
%269 = arith.constant 1 : i32
%271 = arith.extsi %269 : i32 to i64
%270 = arith.addi %268, %271 : i64
llvm.store %270, %247 : i64, !llvm.ptr
cf.br ^bb30
^bb32:
%272 = arith.constant 0 : i32
%273 = arith.extsi %272 : i32 to i64
%274 = llvm.mlir.constant(1 : i64) : i64
%275 = llvm.alloca %274 x i64 : (i64) -> !llvm.ptr
llvm.store %273, %275 : i64, !llvm.ptr
%276 = arith.constant 0 : i32
%277 = arith.extsi %276 : i32 to i64
%278 = llvm.mlir.constant(1 : i64) : i64
%279 = llvm.alloca %278 x i64 : (i64) -> !llvm.ptr
llvm.store %277, %279 : i64, !llvm.ptr
%280 = arith.constant 0 : i32
%281 = arith.extsi %280 : i32 to i64
llvm.store %281, %247 : i64, !llvm.ptr
cf.br ^bb33
^bb33:
%282 = llvm.load %247 : !llvm.ptr -> i64
%283 = llvm.load %187 : !llvm.ptr -> i64
%284 = arith.constant 1 : i32
%286 = arith.extsi %284 : i32 to i64
%285 = arith.subi %283, %286 : i64
%287 = arith.cmpi slt, %282, %285 : i64
cf.cond_br %287, ^bb34, ^bb35
^bb34:
%288 = arith.constant 1 : i32
%290 = llvm.load %239 : !llvm.ptr -> !llvm.ptr
%291 = llvm.load %247 : !llvm.ptr -> i64
%292 = llvm.getelementptr %290[%291] : (!llvm.ptr, i64) -> !llvm.ptr, i64
%289 = llvm.load %292 : !llvm.ptr -> i64
%294 = arith.extsi %288 : i32 to i64
%293 = arith.subi %294, %289 : i64
%296 = llvm.load %239 : !llvm.ptr -> !llvm.ptr
%297 = llvm.load %247 : !llvm.ptr -> i64
%298 = arith.constant 1 : i32
%300 = arith.extsi %298 : i32 to i64
%299 = arith.addi %297, %300 : i64
%301 = llvm.getelementptr %296[%299] : (!llvm.ptr, i64) -> !llvm.ptr, i64
%295 = llvm.load %301 : !llvm.ptr -> i64
%302 = arith.subi %293, %295 : i64
%303 = llvm.load %275 : !llvm.ptr -> i64
%304 = arith.addi %303, %302 : i64
%305 = llvm.mlir.constant(1 : i64) : i64
%306 = llvm.alloca %305 x i64 : (i64) -> !llvm.ptr
llvm.store %304, %306 : i64, !llvm.ptr
%307 = llvm.load %279 : !llvm.ptr -> i64
%308 = llvm.mlir.constant(1 : i64) : i64
%309 = llvm.alloca %308 x i64 : (i64) -> !llvm.ptr
llvm.store %307, %309 : i64, !llvm.ptr
%310 = llvm.load %306 : !llvm.ptr -> i64
%311 = llvm.load %279 : !llvm.ptr -> i64
%312 = arith.cmpi sgt, %310, %311 : i64
cf.cond_br %312, ^bb36, ^bb37
^bb36:
%313 = llvm.load %306 : !llvm.ptr -> i64
llvm.store %313, %309 : i64, !llvm.ptr
cf.br ^bb38
^bb37:
cf.br ^bb38
^bb38:
%314 = llvm.load %279 : !llvm.ptr -> i64
llvm.store %314, %275 : i64, !llvm.ptr
%315 = llvm.load %309 : !llvm.ptr -> i64
llvm.store %315, %279 : i64, !llvm.ptr
%316 = llvm.load %247 : !llvm.ptr -> i64
%317 = arith.constant 1 : i32
%319 = arith.extsi %317 : i32 to i64
%318 = arith.addi %316, %319 : i64
llvm.store %318, %247 : i64, !llvm.ptr
cf.br ^bb33
^bb35:
%320 = llvm.load %279 : !llvm.ptr -> i64
%321 = arith.constant 1 : i32
%323 = llvm.load %239 : !llvm.ptr -> !llvm.ptr
%324 = llvm.load %187 : !llvm.ptr -> i64
%325 = arith.constant 1 : i32
%327 = arith.extsi %325 : i32 to i64
%326 = arith.subi %324, %327 : i64
%328 = llvm.getelementptr %323[%326] : (!llvm.ptr, i64) -> !llvm.ptr, i64
%322 = llvm.load %328 : !llvm.ptr -> i64
%330 = arith.extsi %321 : i32 to i64
%329 = arith.subi %330, %322 : i64
%332 = llvm.load %239 : !llvm.ptr -> !llvm.ptr
%333 = arith.constant 0 : i32
%334 = arith.extsi %333 : i32 to i64
%335 = llvm.getelementptr %332[%334] : (!llvm.ptr, i64) -> !llvm.ptr, i64
%331 = llvm.load %335 : !llvm.ptr -> i64
%336 = arith.subi %329, %331 : i64
%337 = llvm.mlir.constant(1 : i64) : i64
%338 = llvm.alloca %337 x i64 : (i64) -> !llvm.ptr
llvm.store %336, %338 : i64, !llvm.ptr
%339 = llvm.load %187 : !llvm.ptr -> i64
%340 = arith.constant 2 : i32
%342 = arith.extsi %340 : i32 to i64
%341 = arith.cmpi sgt, %339, %342 : i64
cf.cond_br %341, ^bb39, ^bb40
^bb39:
%343 = arith.constant 0 : i32
%344 = arith.extsi %343 : i32 to i64
llvm.store %344, %275 : i64, !llvm.ptr
%345 = arith.constant 0 : i32
%346 = arith.extsi %345 : i32 to i64
llvm.store %346, %279 : i64, !llvm.ptr
%347 = arith.constant 1 : i32
%348 = arith.extsi %347 : i32 to i64
llvm.store %348, %247 : i64, !llvm.ptr
cf.br ^bb42
^bb42:
%349 = llvm.load %247 : !llvm.ptr -> i64
%350 = llvm.load %187 : !llvm.ptr -> i64
%351 = arith.constant 2 : i32
%353 = arith.extsi %351 : i32 to i64
%352 = arith.subi %350, %353 : i64
%354 = arith.cmpi slt, %349, %352 : i64
cf.cond_br %354, ^bb43, ^bb44
^bb43:
%355 = arith.constant 1 : i32
%357 = llvm.load %239 : !llvm.ptr -> !llvm.ptr
%358 = llvm.load %247 : !llvm.ptr -> i64
%359 = llvm.getelementptr %357[%358] : (!llvm.ptr, i64) -> !llvm.ptr, i64
%356 = llvm.load %359 : !llvm.ptr -> i64
%361 = arith.extsi %355 : i32 to i64
%360 = arith.subi %361, %356 : i64
%363 = llvm.load %239 : !llvm.ptr -> !llvm.ptr
%364 = llvm.load %247 : !llvm.ptr -> i64
%365 = arith.constant 1 : i32
%367 = arith.extsi %365 : i32 to i64
%366 = arith.addi %364, %367 : i64
%368 = llvm.getelementptr %363[%366] : (!llvm.ptr, i64) -> !llvm.ptr, i64
%362 = llvm.load %368 : !llvm.ptr -> i64
%369 = arith.subi %360, %362 : i64
%370 = llvm.load %275 : !llvm.ptr -> i64
%371 = arith.addi %370, %369 : i64
%372 = llvm.mlir.constant(1 : i64) : i64
%373 = llvm.alloca %372 x i64 : (i64) -> !llvm.ptr
llvm.store %371, %373 : i64, !llvm.ptr
%374 = llvm.load %279 : !llvm.ptr -> i64
%375 = llvm.mlir.constant(1 : i64) : i64
%376 = llvm.alloca %375 x i64 : (i64) -> !llvm.ptr
llvm.store %374, %376 : i64, !llvm.ptr
%377 = llvm.load %373 : !llvm.ptr -> i64
%378 = llvm.load %279 : !llvm.ptr -> i64
%379 = arith.cmpi sgt, %377, %378 : i64
cf.cond_br %379, ^bb45, ^bb46
^bb45:
%380 = llvm.load %373 : !llvm.ptr -> i64
llvm.store %380, %376 : i64, !llvm.ptr
cf.br ^bb47
^bb46:
cf.br ^bb47
^bb47:
%381 = llvm.load %279 : !llvm.ptr -> i64
llvm.store %381, %275 : i64, !llvm.ptr
%382 = llvm.load %376 : !llvm.ptr -> i64
llvm.store %382, %279 : i64, !llvm.ptr
%383 = llvm.load %247 : !llvm.ptr -> i64
%384 = arith.constant 1 : i32
%386 = arith.extsi %384 : i32 to i64
%385 = arith.addi %383, %386 : i64
llvm.store %385, %247 : i64, !llvm.ptr
cf.br ^bb42
^bb44:
%387 = llvm.load %279 : !llvm.ptr -> i64
%388 = arith.addi %336, %387 : i64
llvm.store %388, %338 : i64, !llvm.ptr
cf.br ^bb41
^bb40:
cf.br ^bb41
^bb41:
%389 = llvm.mlir.constant(1 : i64) : i64
%390 = llvm.alloca %389 x i64 : (i64) -> !llvm.ptr
llvm.store %320, %390 : i64, !llvm.ptr
%391 = llvm.load %338 : !llvm.ptr -> i64
%392 = llvm.load %390 : !llvm.ptr -> i64
%393 = arith.cmpi sgt, %391, %392 : i64
cf.cond_br %393, ^bb48, ^bb49
^bb48:
%394 = llvm.load %338 : !llvm.ptr -> i64
llvm.store %394, %390 : i64, !llvm.ptr
cf.br ^bb50
^bb49:
cf.br ^bb50
^bb50:
%395 = llvm.load %390 : !llvm.ptr -> i64
%396 = arith.constant 0 : i32
%398 = arith.extsi %396 : i32 to i64
%397 = arith.cmpi slt, %395, %398 : i64
cf.cond_br %397, ^bb51, ^bb52
^bb51:
%399 = arith.constant 0 : i32
%400 = arith.extsi %399 : i32 to i64
llvm.store %400, %390 : i64, !llvm.ptr
cf.br ^bb53
^bb52:
cf.br ^bb53
^bb53:
%401 = llvm.load %156 : !llvm.ptr -> i64
%402 = llvm.load %243 : !llvm.ptr -> i64
%403 = arith.addi %401, %402 : i64
%404 = llvm.load %390 : !llvm.ptr -> i64
%405 = arith.addi %403, %404 : i64
llvm.store %405, %156 : i64, !llvm.ptr
%407 = llvm.load %239 : !llvm.ptr -> !llvm.ptr
func.call @free(%407) : (!llvm.ptr) -> ()
cf.br ^bb29
^bb29:
%409 = llvm.load %183 : !llvm.ptr -> !llvm.ptr
func.call @free(%409) : (!llvm.ptr) -> ()
cf.br ^bb23
^bb22:
cf.br ^bb23
^bb23:
%410 = llvm.load %160 : !llvm.ptr -> i64
%411 = arith.constant 1 : i32
%413 = arith.extsi %411 : i32 to i64
%412 = arith.addi %410, %413 : i64
llvm.store %412, %160 : i64, !llvm.ptr
cf.br ^bb18
^bb20:
%415 = llvm.load %4 : !llvm.ptr -> !llvm.ptr
func.call @free(%415) : (!llvm.ptr) -> ()
%417 = llvm.load %9 : !llvm.ptr -> !llvm.ptr
func.call @free(%417) : (!llvm.ptr) -> ()
%419 = llvm.load %47 : !llvm.ptr -> !llvm.ptr
func.call @free(%419) : (!llvm.ptr) -> ()
%421 = llvm.load %52 : !llvm.ptr -> !llvm.ptr
func.call @free(%421) : (!llvm.ptr) -> ()
%423 = llvm.load %57 : !llvm.ptr -> !llvm.ptr
func.call @free(%423) : (!llvm.ptr) -> ()
%425 = llvm.load %152 : !llvm.ptr -> !llvm.ptr
func.call @free(%425) : (!llvm.ptr) -> ()
%426 = llvm.load %156 : !llvm.ptr -> i64
func.return %426 : i64
}
func.func @main() -> i32 {
%427 = arith.constant 0 : i32
%428 = arith.extsi %427 : i32 to i64
%429 = llvm.mlir.constant(1 : i64) : i64
%430 = llvm.alloca %429 x i64 : (i64) -> !llvm.ptr
llvm.store %428, %430 : i64, !llvm.ptr
%431 = arith.constant 1 : i32
%432 = arith.extsi %431 : i32 to i64
%433 = llvm.mlir.constant(1 : i64) : i64
%434 = llvm.alloca %433 x i64 : (i64) -> !llvm.ptr
llvm.store %432, %434 : i64, !llvm.ptr
cf.br ^bb54
^bb54:
%435 = llvm.load %434 : !llvm.ptr -> i64
%436 = arith.constant 100 : i32
%438 = arith.extsi %436 : i32 to i64
%437 = arith.cmpi sle, %435, %438 : i64
cf.cond_br %437, ^bb55, ^bb56
^bb55:
%439 = llvm.load %430 : !llvm.ptr -> i64
%441 = arith.constant 100000 : i32
%442 = llvm.load %434 : !llvm.ptr -> i64
%444 = arith.extsi %441 : i32 to i64
%443 = arith.addi %444, %442 : i64
%440 = func.call @drifting_subset_size(%443) : (i64) -> i64
%445 = arith.addi %439, %440 : i64
llvm.store %445, %430 : i64, !llvm.ptr
%446 = llvm.load %434 : !llvm.ptr -> i64
%447 = arith.constant 1 : i32
%449 = arith.extsi %447 : i32 to i64
%448 = arith.addi %446, %449 : i64
llvm.store %448, %434 : i64, !llvm.ptr
cf.br ^bb54
^bb56:
%450 = llvm.mlir.addressof @str_0 : !llvm.ptr
%451 = llvm.load %430 : !llvm.ptr -> i64
%452 = llvm.call @printf(%450, %451) vararg(!llvm.func<i32 (ptr, ...)>) : (!llvm.ptr, i64) -> i32
%453 = arith.constant 0 : i32
func.return %453 : i32
}
}