int count(int n) {
    int res = 0;
    for (int i = 1; i <= n; i *= 2) {
        for (int j = 1; j <= n; j++) {
            res++;
        }
    }
    return res;
}