結果

問題 No.3389 k-Days Later
コンテスト
ユーザー amesyu
提出日時 2025-11-28 22:03:46
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 539 ms / 2,000 ms
コード長 4,222 bytes
コンパイル時間 3,584 ms
コンパイル使用メモリ 289,864 KB
実行使用メモリ 7,848 KB
最終ジャッジ日時 2025-11-28 22:05:02
合計ジャッジ時間 22,519 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 28
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include <bits/stdc++.h>
#pragma GCC target("avx2")
/*
#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")
*/
#pragma GCC target("prefer-vector-width=512")
#pragma GCC optimize("Ofast,unroll-loops")

using namespace std;
using i32 = int;
using i64 = long long;
using f32 = float;
using f64 = double;
using f80 = long double;

#define rep(i, n) for(i64 i = 0; i < (n); i++)
#define all(x) (x).begin(), (x).end()

template<class T, class U>
ostream& operator<<(ostream& os, const pair<T, U>& p);
template<class T>
ostream& operator<<(ostream& os, const vector<T>& v);

template<class T, class U>
ostream& operator<<(ostream& os, const pair<T, U>& p) {
    os << "(" << p.first << ", " << p.second << ")";
    return os;
}

template<class T>
istream& operator>>(istream& is, vector<T>& v) {
    for(int i = 0; i < (int)v.size(); i++) {
        is >> v[i];
    }
    return is;
}

template<class T>
ostream& operator<<(ostream& os, const vector<T>& v) {
    for(int i = 0; i < (int)v.size(); i++) {
        if(i > 0) os << " ";
        os << v[i];
    }
    return os;
}

void YesNo(bool yn) {
    if(yn) cout << "Yes" << endl;
    else cout << "No" << endl;
}

void YesNoExit(bool yn) {
    YesNo(yn);
    exit(0);
}

template<class T, class U>
bool chmin(T &x, U y) {
    if(y < x) {
        x = y;
        return true;
    }
    return false;
}

template<class T, class U>
bool chmax(T &x, U y) {
    if(y > x) {
        x = y;
        return true;
    }
    return false;
}

struct UnionFind {
  vector<int> data;

  UnionFind() = default;

  explicit UnionFind(size_t sz) : data(sz, -1) {}

  bool unite(int x, int y) {
    x = find(x), y = find(y);
    if (x == y) return false;
    if (data[x] > data[y]) swap(x, y);
    data[x] += data[y];
    data[y] = x;
    return true;
  }

  int find(int k) {
    if (data[k] < 0) return (k);
    return data[k] = find(data[k]);
  }

  int size(int k) { return -data[find(k)]; }

  bool same(int x, int y) { return find(x) == find(y); }

  vector<vector<int> > groups() {
    int n = (int)data.size();
    vector<vector<int> > ret(n);
    for (int i = 0; i < n; i++) {
      ret[find(i)].emplace_back(i);
    }
    ret.erase(remove_if(begin(ret), end(ret),
                        [&](const vector<int>& v) { return v.empty(); }),
              end(ret));
    return ret;
  }
};

void out() { cout << '\n'; }
template <class T, class... Ts>
void out(const T& a, const Ts&... b) {
    cout << a;
    (cout << ... << (cout << ' ', b));
    cout << '\n';
}

vector<i32> dx4 = {0, 1, 0, -1};
vector<i32> dy4 = {1, 0, -1, 0};

const i64 MOD = 998244353;
const i32 i32_inf = 1e9;
const i64 i64_inf = 1e18;

struct Eratosthenes {
    int _n;
    vector<int> prv;
    Eratosthenes() {}
    Eratosthenes(int n) : _n(n + 1), prv(_n, 0) {
        prv[0] = -1;
        prv[1] = -1;
        for(int i = 2; i < _n; ++i) {
            if(prv[i]) continue;
            for(int k = 2; i * k < _n; ++k) {
                prv[i * k] = k;
            }
        }
    }

    bool isPrime(int x) {
        return prv[x] == 0;
    }

    vector<int> get(int x) {
        assert(2 <= x && x < _n);
        vector<int> ret;
        while(prv[x]) {
            ret.emplace_back(x / prv[x]);
            x = prv[x];
        }
        ret.emplace_back(x);
        return ret;
    }
};

void solve() {
    i32 n; cin >> n;
    vector<i64> d(n);
    cin >> d;

    i64 s = 0;
    rep(i, n) s += d[i];

    vector<i64> csum(1, 0);
    for(int i = 0; i < 2 * n; ++i) {
        csum.push_back(csum[i] + d[i % n]);
    }

    i32 q; cin >> q;
    while(q--) {
        i64 y, m, d, k; cin >> y >> m >> d >> k;
        m--;
        d--;

        k += d;
        d = 0;

        y += k / s;
        k %= s;

        i32 ok = m;
        i32 ng = m + n;
        while(ng - ok > 1) {
            i64 x = (ok + ng) >> 1;
            if(csum[x] - csum[m] <= k) {
                ok = x; 
            } else {
                ng = x;
            }
        }

        d = k - (csum[ok] - csum[m]);
        if(ok >= n) y++;
        m = ok % n;

        m++;
        d++;

        out(y, m, d);
    }

}

int main() {
    int TESTCASE = 1;
    // cin >> TESTCASE;
    while(TESTCASE--) {
        solve();
    }
}
0