#include #include #include // AVX2 header #pragma GCC target("avx2") using namespace std; // Aligned allocation for SIMD efficiency long long* allocate_aligned(int n) { void* ptr = nullptr; if (posix_memalign(&ptr, 32, n * sizeof(long long)) != 0) return nullptr; return (long long*)ptr; } int main() { ios::sync_with_stdio(false); cin.tie(nullptr); int N; cin >> N; long long* arr = allocate_aligned(N + 1); for (int i = 1; i <= N; ++i) { cin >> arr[i]; } int Q; cin >> Q; while (Q--) { int type; cin >> type; if (type == 1) { int pos; long long val; cin >> pos >> val; arr[pos] = val; } else { int l, r; cin >> l >> r; long long total_sum = 0; int i = l; // AVX2 logic: Process 4 long longs (64-bit) at a time __m256i vsum = _mm256_setzero_si256(); // Align start to 4-element boundary if necessary or just handle chunks for (; i + 3 <= r; i += 4) { __m256i vdata = _mm256_loadu_si256((__m256i*)&arr[i]); vsum = _mm256_add_epi64(vsum, vdata); } // Horizontal sum of the 256-bit register long long temp[4]; _mm256_storeu_si256((__m256i*)temp, vsum); total_sum = temp[0] + temp[1] + temp[2] + temp[3]; // Handle remainder for (; i <= r; ++i) { total_sum += arr[i]; } cout << total_sum << "\n"; } } free(arr); return 0; }

Submission #96

Source Code