結果

問題 No.1093 区間の和 / Sum of Range
ユーザー WarToks
提出日時 2020-07-04 22:15:03
言語 C++17(clang)
(17.0.6 + boost 1.87.0)
結果
AC  
実行時間 13 ms / 2,000 ms
コード長 1,750 bytes
コンパイル時間 746 ms
コンパイル使用メモリ 121,216 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2024-11-30 19:00:11
合計ジャッジ時間 2,549 ms
ジャッジサーバーID
(参考情報)
judge1 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 1
other AC * 36
権限があれば一括ダウンロードができます

ソースコード

diff #

#pragma GCC optimize("O3", "unroll-loops")
#pragma GCC target("avx")

#include <stdio.h>
#include <algorithm>

static constexpr int buffer_size = 1 << 17;
static char buffer[buffer_size];
static int pos = buffer_size;

inline char get_char(void){
    if(pos == buffer_size){ fread_unlocked(buffer, 1, buffer_size, stdin); pos = 0; }
    return buffer[pos++];
}

inline int getint(void){
    int res = 0; char c = get_char();
    if(c == '-'){
        while((c = get_char() - '0') >= 0) res = 10 * res + c;
        return -res;
    }
    res = c - '0';
    while((c = get_char() - '0') >= 0) res = 10 * res + c;
    return res;
}

inline void put_char(const char c){
    if(pos == buffer_size){ fwrite_unlocked(buffer, 1, buffer_size, stdout); pos = 0; }
    buffer[pos++] = c;
}


char buffer2[20];
inline void println(int x){
    if(x){
        int idx = 0; do{ buffer2[idx++] = x % 10; x /= 10; } while(x);
        while(idx) put_char( buffer2[--idx] + '0');
    }
    else put_char('0');
    put_char('\n');
}


constexpr int MAX_N = 100000;
constexpr int MAX_Q = 100000;
int A[MAX_N], S[MAX_N], X[MAX_N];


int main(void){
    const int n = getint(), k = getint();
    for(int i = 0; i < n; ++i) A[i] = getint();{
        int val = 0; for(int i = 0; i < k; ++i) val += A[i];
        for(int i = 0, j = k; j < n; ++i, ++j){
            S[i] = val;
            val += A[j] - A[i];
        }
        S[n - k] = val;
    }
    int* T = S + n - k + 1;
    std::sort(S, T);

    const int q = getint();
    for(int i = 0; i < q; ++i) X[i] = getint();
    
    pos = 0;
    for(int i = 0; i < q; ++i){
        const int res = std::upper_bound(S, T, X[i]) - S;
        println(res);
    }
    
    fwrite_unlocked(buffer, 1, pos, stdout);
    return 0;
}
0