Zxc200611/pocketpy
1
0
Fork 0
mirror of https://github.com/pocketpy/pocketpy synced 2025-12-07 02:30:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

some optimize

Browse Source
This commit is contained in:
blueloveTH 2024-06-16 00:21:53 +08:00
parent e765e9f9bd
commit 563ae74a19
2 changed files with 15 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
include/pocketpy/common/algorithm.h
View File

@ -13,7 +13,7 @@ extern "C" {
T* __first = ptr; \
int __len = count; \
while(__len != 0) { \
int __l2 = (int)((unsigned int)__len / 2); \
int __l2 = __len >> 1; \
T* __m = __first + __l2; \
if(less((*__m), (key))) { \
__first = ++__m; \

21
include/pocketpy/xmacros/smallmap.h
View File

@ -64,14 +64,21 @@ void SMALLMAP_METHOD(set)(SMALLMAP* self, K key, V value) {
}
V* SMALLMAP_METHOD(try_get)(const SMALLMAP* self, K key) {
int index;
c11__lower_bound(KV, self->data, self->count, key, less, &index);
KV* it = c11__at(KV, self, index);
if(index != self->count && equal(it->key, key)) {
return &it->value;
} else {
return NULL;
// use `bsearch` which is faster than `lower_bound`
int low = 0;
int high = self->count - 1;
KV* a = self->data;
while(low <= high){
int mid = (low + high) / 2;
if(equal(a[mid].key, key)){
return &a[mid].value;
} else if(less(a[mid], key)){
low = mid + 1;
} else {
high = mid - 1;
}
}
return NULL;
}
V SMALLMAP_METHOD(get)(const SMALLMAP* self, K key, V default_value) {