a/SDC
1
1
Fork 0
Code Issues 1 Pull requests Projects Releases Wiki Activity

Clean up code and fix a typo in segmenttree.c

Browse source
This commit is contained in:
Anthony Wang 2022-04-19 15:13:09 -05:00
parent b95cdd56c2
commit 03b09cc02a
Signed by: a
GPG key ID: BC96B00AEC5F2D76
2 changed files with 9 additions and 29 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

34
cd.c
View file

@ -33,35 +33,23 @@ int main(int argc, char* argv[]) {
/* Connect to db */
sqlite3 *db;
int rc = sqlite3_open(file, &db);
if (rc) {
fprintf(stderr, "Can't open database: %s\n", sqlite3_errmsg(db));
abort();
}
sqlite3_open(file, &db);
/* Get number of cards */
sqlite3_stmt *stmt;
rc = sqlite3_prepare_v3(db, "SELECT COUNT(*) FROM cards", -1, 0, &stmt, NULL);
if (rc) {
fprintf(stderr, "Database error: %s\n", sqlite3_errmsg(db));
abort();
}
sqlite3_prepare_v3(db, "SELECT COUNT(*) FROM cards", -1, 0, &stmt, NULL);
sqlite3_step(stmt);
int N = sqlite3_column_int(stmt, 0);
sqlite3_finalize(stmt);
/* Get card weights */
rc = sqlite3_prepare_v3(db, "SELECT weight FROM cards", -1, 0, &stmt, NULL);
if (rc) {
fprintf(stderr, "Database error: %s\n", sqlite3_errmsg(db));
abort();
}
seg = malloc(4 * N * sizeof(int));
sqlite3_prepare_v3(db, "SELECT weight FROM cards", -1, 0, &stmt, NULL);
seg = (int*)malloc(4 * N * sizeof(int));
build(stmt, 0, N - 1, 1);
sqlite3_finalize(stmt);
if (verbose) {
for (int i = 0; i < N; i++) {
for (int i = 0; i < 4 * N; i++) {
printf("%d ", seg[i]);
}
printf("\n");
@ -82,11 +70,7 @@ int main(int argc, char* argv[]) {
}
/* Get card contents from database */
rc = sqlite3_prepare_v3(db, "SELECT key, val FROM cards WHERE idx=?", -1, 0, &stmt, NULL);
if (rc) {
fprintf(stderr, "Database error: %s\n", sqlite3_errmsg(db));
abort();
}
sqlite3_prepare_v3(db, "SELECT key, val FROM cards WHERE idx=?", -1, 0, &stmt, NULL);
sqlite3_bind_int(stmt, 1, i);
sqlite3_step(stmt);
char *key, *val;
@ -114,11 +98,7 @@ int main(int argc, char* argv[]) {
/* Update segment tree and database */
update(i, w, 0, N - 1, 1);
rc = sqlite3_prepare_v3(db, "UPDATE cards SET weight=? WHERE idx=?", -1, 0, &stmt, NULL);
if (rc) {
fprintf(stderr, "Database error: %s\n", sqlite3_errmsg(db));
abort();
}
sqlite3_prepare_v3(db, "UPDATE cards SET weight=? WHERE idx=?", -1, 0, &stmt, NULL);
sqlite3_bind_int(stmt, 1, w);
sqlite3_bind_int(stmt, 2, i);
sqlite3_step(stmt);

4
segmenttree.c
View file

@ -6,7 +6,7 @@ int * seg;
void build(sqlite3_stmt *stmt, int l, int r, int n) {
if (l == r) {
sqlite3_step(stmt);
seg[l] = sqlite3_column_int(stmt, 0);
seg[n] = sqlite3_column_int(stmt, 0);
return;
}
int m = l + r >> 1;
@ -45,4 +45,4 @@ void query(int *res, int v, int l, int r, int n) {
else {
query(res, v - seg[n << 1], m + 1, r, n << 1 | 1);
}
}
}