結果

問題 No.3097 Azuki Kurai
ユーザー RiRinbaru
提出日時 2025-04-05 12:40:05
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 251 ms / 4,000 ms
コード長 2,770 bytes
コンパイル時間 4,596 ms
コンパイル使用メモリ 229,156 KB
実行使用メモリ 13,568 KB
最終ジャッジ日時 2025-04-05 12:40:19
合計ジャッジ時間 12,422 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 32
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#define rep(i, p, n) for (ll i = p; i < (ll)(n); i++)
#define rep2(i, p, n) for (ll i = p; i >= (ll)(n); i--)
using namespace std;
using ll = long long;
using ld = long double;
const double pi = 3.141592653589793;
const long long inf = 2 * 1e9;
const long long linf = 4 * 1e18;
const ll mod1 = 1000000007;
const ll mod2 = 998244353;
template <class T>
inline bool chmax(T &a, T b)
{
    if (a < b)
    {
        a = b;
        return 1;
    }
    return 0;
}
template <class T>
inline bool chmin(T &a, T b)
{
    if (a > b)
    {
        a = b;
        return 1;
    }
    return 0;
}

// atcoder
#include <atcoder/all>
using namespace atcoder;
using mint1 = modint1000000007;
using mint2 = modint998244353;

vector<pair<ll, ll>> base = {{-1, 0}, {1, 0}, {0, -1}, {0, 1}};

int main()
{

    //////////////////
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    cout.tie(nullptr);
    //////////////////

    ll N, M, K;
    cin >> N >> M >> K;
    vector<ll> A(N);
    rep(i, 0, N)
    {
        cin >> A.at(i);
    }
    vector<vector<ll>> dp(M + 1, vector<ll>((1 << N), linf));
    rep(i, 0, (1 << N))
    {
        dp.at(0).at(i) = 0;
        rep(j, 0, N)
        {
            if (!(i & (1 << j)))
            {
                dp.at(0).at(i) += A.at(j);
            }
        }
    }
    ll now, temp;
    vector<vector<ll>> pl(1 << N);
    vector<vector<vector<ll>>> mae(N, vector<vector<ll>>((1 << N)));
    rep(i, 0, (1 << N))
    {
        for (ll c = i; c; c = (c - 1) & i)
        {
            pl.at(i).push_back((__builtin_popcount(i) - __builtin_popcount(c)) * K);
        }
        pl.at(i).push_back((__builtin_popcount(i) - __builtin_popcount(0)) * K);
    }
    rep(p, 0, N)
    {
        rep(i, 0, (1 << N))
        {
            for (ll c = i; c; c = (c - 1) & i)
            {
                now = i;
                now |= ((c * 2) - ((c * 2) & (1 << N)));
                now |= (((c * 2) & (1 << N)) >> N);
                now |= (c / 2);
                now |= (c % 2) * (1 << (N - 1));
                now -= (now & (1 << p));
                mae.at(p).at(i).push_back(now);
            }
            now = i;
            now -= (now & (1 << p));
            mae.at(p).at(i).push_back(now);
        }
    }
    rep(p, 0, M)
    {
        ll B;
        cin >> B;
        B--;
        rep(i, 0, (1 << N))
        {
            ll F = 0;
            for (ll c = i; c; c = (c - 1) & i)
            {
                chmin(dp.at(p + 1).at(mae.at(B).at(i).at(F)), dp.at(p).at(i) + pl.at(i).at(F));
                F++;
            }
            chmin(dp.at(p + 1).at(mae.at(B).at(i).at(F)), dp.at(p).at(i) + pl.at(i).at(F));
        }
    }
    rep(i, 0, M)
    {
        cout << dp.at(i + 1).at(0) << endl;
    }
}
0