제출 #1356094

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1356094		Joshua_Andersson	사탕 분배 (IOI21_candies)	C++20	100 / 100	4003 ms	7476 KiB

candies

#include "candies.h"

#pragma GCC optimize("O3")
#pragma GCC target("avx2")
#include <vector>
#include <iostream>
using namespace std;

using vi = vector<int>;
#define sz(x) ((int)x.size())
#define rep(i,n) for (int i = 0; i < (n); i++)
#define repp(i,a,n) for (int i = (a); i < (n); i++)

std::vector<int> distribute_candies(std::vector<int> c, std::vector<int> l,
    std::vector<int> r, std::vector<int> v) {

    int orig_n = sz(c);
    const int B = 1000;
    int n = sz(c);
    while (n%B!=0) n++;
    while (sz(c) < n) c.push_back(0);

    vi ans(n);

    for (int b = 0; b < n; b+=B) {
        rep(q, sz(v)) {
            int lo = max(b, l[q]);
            int hi = min(b+B, r[q]+1);
            
            int addend = v[q];
            if (v[q]>0) {
                #pragma GCC ivdep
                #pragma GCC unroll 8
                repp(i,lo,hi) {
                    ans[i]+=addend;
                    ans[i]=min(c[i],ans[i]);
                }
            }
            else {
                #pragma GCC ivdep
                #pragma GCC unroll 8
                repp(i,lo,hi) {
                    ans[i]+=addend;
                    ans[i]=max(0,ans[i]);
                }
            }
        }
    }

    while (sz(ans)>orig_n) ans.pop_back();
    return ans;
}

• 서브태스크 1: 3 / 3
• 서브태스크 2: 8 / 8
• 서브태스크 3: 27 / 27
• 서브태스크 4: 29 / 29
• 서브태스크 5: 33 / 33