Yukicoder No.489 株に挑戦
yukicoderではavxが使える。嘘解法力の上昇を感じる。
solution
愚直。$O(ND)$。
コンパイラさんがお仕事しやすいように、一度$\max \{ x_k \mid j \le k \le j + D \}$だけ単体で求めてから復元するように書くとよい。
implementation
pragma付けるだけでよろしくしてくれて驚き。
objdumpするとXMMWORD
なvpmaxsd
命令が吐かれていることが確認できた。
#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;
}