Yukicoder No.489 株に挑戦

,

https://yukicoder.me/problems/no/489

yukicoderではavxが使える。嘘解法力の上昇を感じる。

solution

愚直。$O(ND)$。

コンパイラさんがお仕事しやすいように、一度$\max \{ x_k \mid j \le k \le j + D \}$だけ単体で求めてから復元するように書くとよい。

implementation

pragma付けるだけでよろしくしてくれて驚き。 objdumpするとXMMWORDvpmaxsd命令が吐かれていることが確認できた。

#pragma GCC optimize ("O3")
#pragma GCC target ("tune=native")
#pragma GCC target ("avx")
#include <cstdio>
#include <algorithm>
#define repeat(i,n) for (int i = 0; (i) < int(n); ++(i))
#define repeat_from(i,m,n) for (int i = (m); (i) < int(n); ++(i))
using ll = long long;
using namespace std;
constexpr int MAX_N = 1e5;
int x[MAX_N];
int main() {
    int n, d, K; scanf("%d%d%d", &n, &d, &K);
    repeat (i,n) scanf("%d", &x[i]);
    int mx = 0, mj = -1;
    repeat (j,n) {
        int acc = 0;
        repeat_from (k,j,min(n,j+d+1)) acc = max(acc, x[k]); // simd
        if (mx < acc - x[j]) {
            mx = acc - x[j];
            mj = j;
        }
    }
    int mk = -1;
    repeat_from (k,mj,min(n,mj+d+1)) {
        if (mx == x[k] - x[mj]) {
            mk = k;
            break;
        }
    }
    printf("%lld\n", mx*(ll)K);
    if (mx) printf("%d %d\n", mj, mk);
    return 0;
}