提出 #68540023


ソースコード 拡げる

//#pragma GCC target("avx2")
//#pragma GCC optimize("O3")
//#pragma GCC optimize("unroll-loops")
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
using pii = pair<int,int>;
using pll = pair<ll,ll>;
using pli = pair<ll,int>;
#define AMARI 998244353
//#define AMARI 1000000007
#define el '\n'
#define El '\n'
#define YESNO(x) ((x) ? "Yes" : "No")
#define YES YESNO(true)
#define NO YESNO(false)
#define EXIT_ANS(x) {cout << (x) << '\n'; return;}
template <typename T> void inline SORT(vector<T> &v){sort(v.begin(),v.end()); return;}
template <typename T> void inline REV(vector<T> &v){reverse(v.begin(),v.end()); return;}
template <typename T> void inline VEC_UNIQ(vector<T> &v){sort(v.begin(),v.end()); v.erase(unique(v.begin(),v.end()),v.end()); return;}
template <typename T> T inline MAX(vector<T> &v){return *max_element(v.begin(),v.end());}
template <typename T> T inline MIN(vector<T> &v){return *min_element(v.begin(),v.end());}
template <typename T> T inline SUM(vector<T> &v){T ans = 0; for(int i = 0; i < (int)v.size(); i++)ans += v[i]; return ans;}
template <typename T> void inline DEC(vector<T> &v){for(int i = 0; i < (int)v.size(); i++)v[i]--; return;}
template <typename T> void inline INC(vector<T> &v){for(int i = 0; i < (int)v.size(); i++)v[i]++; return;}
void inline TEST(void){cerr << "TEST" << endl; return;}
template <typename T> bool inline chmin(T &x,T y){
    if(x > y){
        x = y;
        return true;
    }
    return false;
}
template <typename T> bool inline chmax(T &x,T y){
    if(x < y){
        x = y;
        return true;
    }
    return false;
}
template <typename T = long long> vector<T> inline get_vec(int n){
    vector<T> ans(n);
    for(int i = 0; i < n; i++)cin >> ans[i];
    return ans;
}
template <typename T> void inline print_vec(vector<T> &vec,bool kaigyou = false){
    int n = (int)vec.size();
    for(int i = 0; i < n; i++){
        cout << vec[i];
        if(kaigyou || i == n - 1)cout << '\n';
        else cout << ' ';
    }
    if(!n)cout << '\n';
    return;
}
template <typename T> void inline debug_vec(vector<T> &vec,bool kaigyou = false){
    int n = (int)vec.size();
    for(int i = 0; i < n; i++){
        cerr << vec[i];
        if(kaigyou || i == n - 1)cerr << '\n';
        else cerr << ' ';
    }
    if(!n)cerr << '\n';
    return;
}
vector<vector<int>> inline get_graph(int n,int m = -1,bool direct = false){
    if(m == -1)m = n - 1;
    vector<vector<int>> g(n);
    while(m--){
        int u,v;
        cin >> u >> v;
        u--; v--;
        g[u].push_back(v);
        if(!direct)g[v].push_back(u);
    }
    return g;
}
template <typename T> vector<vector<T>> inline get_weighted_graph(int n,int m = -1,bool direct = false){
    if(m == -1)m = n - 1;
    vector<vector<pair<T,int>>> g(n);
    while(m--){
        int u,v;
        cin >> u >> v;
        u--; v--;
        ll w; cin >> w;
        g[u].push_back(pair(w,v));
        if(!direct)g[v].push_back(pair(w,u));
    }
    return g;
}

vector<int> make_inv(vector<int> p){
    int n = (int)p.size();
    vector<int> ans(n);
    for(int i = 0; i < n; i++)ans[p[i]] = i;
    return ans;
}

#define MULTI_TEST_CASE false
void solve(void){
    //問題を見たらまず「この問題設定から言えること」をいっぱい言う
    //よりシンプルな問題に言い換えられたら、言い換えた先の問題を自然言語ではっきりと書く
    //複数の解法のアイデアを思いついた時は全部メモしておく
    //g++ -D_GLIBCXX_DEBUG -Wall -O2 e.cpp -o o
    int n,m,l;
    cin >> n >> m >> l;
    vector<int> a =get_vec<int>(n);

    vector<vector<int>> vec(l);

    for(int i = 0; i < n; i++){
        vec[i % l].push_back(a[i]);
    }

    //cost[i][j] = i番目について、 mod m で j にするのに必要なコストはいくらか
    vector<vector<int>> cost(l,vector<int>(m));

    for(int i = 0; i < l; i++){
        SORT(vec[i]);
        for(int j = 0; j < m; j++){
            for(int k = 0; k < vec[i].size(); k++){
                if(vec[i][k] == j)continue;
                int dif = j - vec[i][k];
                if(dif < 0)dif += m;
                cost[i][j] += dif;
            }
        }
    }

    //for(int i = 0; i < l; i++)debug_vec(cost)

    vector<vector<int>> dp(l,vector<int>(m,INT_MAX));
    for(int i = 0; i < m; i++)dp[0][i] = cost[0][i];

    for(int i = 1; i < l; i++){
        for(int j = 0; j < m; j++){
            for(int k = 0; k < m; k++){
                chmin(dp[i][(j + k) % m],dp[i - 1][j] + cost[i][k]);
            }
        }
    }
    cout << dp[l - 1][0] << el;
    return;
}

void calc(void){
    return;
}

signed main(void){
    cin.tie(nullptr);
    ios::sync_with_stdio(false);
    calc();
    int t = 1;
    if(MULTI_TEST_CASE)cin >> t;
    while(t--){
        solve();
    }
    return 0;
}

提出情報

提出日時
問題 E - Subarray Sum Divisibility
ユーザ ococonomy1
言語 C++ 17 (gcc 12.2)
得点 475
コード長 5004 Byte
結果 AC
実行時間 218 ms
メモリ 5168 KiB

コンパイルエラー

Main.cpp: In function ‘void solve()’:
Main.cpp:121:30: warning: comparison of integer expressions of different signedness: ‘int’ and ‘std::vector<int>::size_type’ {aka ‘long unsigned int’} [-Wsign-compare]
  121 |             for(int k = 0; k < vec[i].size(); k++){
      |                            ~~^~~~~~~~~~~~~~~

ジャッジ結果

セット名 Sample All
得点 / 配点 0 / 0 475 / 475
結果
AC × 2
AC × 37
セット名 テストケース
Sample sample00.txt, sample01.txt
All sample00.txt, sample01.txt, testcase00.txt, testcase01.txt, testcase02.txt, testcase03.txt, testcase04.txt, testcase05.txt, testcase06.txt, testcase07.txt, testcase08.txt, testcase09.txt, testcase10.txt, testcase11.txt, testcase12.txt, testcase13.txt, testcase14.txt, testcase15.txt, testcase16.txt, testcase17.txt, testcase18.txt, testcase19.txt, testcase20.txt, testcase21.txt, testcase22.txt, testcase23.txt, testcase24.txt, testcase25.txt, testcase26.txt, testcase27.txt, testcase28.txt, testcase29.txt, testcase30.txt, testcase31.txt, testcase32.txt, testcase33.txt, testcase34.txt
ケース名 結果 実行時間 メモリ
sample00.txt AC 1 ms 3508 KiB
sample01.txt AC 1 ms 3388 KiB
testcase00.txt AC 1 ms 3636 KiB
testcase01.txt AC 1 ms 3496 KiB
testcase02.txt AC 1 ms 3516 KiB
testcase03.txt AC 1 ms 3448 KiB
testcase04.txt AC 108 ms 4172 KiB
testcase05.txt AC 207 ms 5168 KiB
testcase06.txt AC 1 ms 3648 KiB
testcase07.txt AC 104 ms 4068 KiB
testcase08.txt AC 211 ms 5100 KiB
testcase09.txt AC 1 ms 3536 KiB
testcase10.txt AC 12 ms 3624 KiB
testcase11.txt AC 218 ms 5120 KiB
testcase12.txt AC 2 ms 3508 KiB
testcase13.txt AC 2 ms 3532 KiB
testcase14.txt AC 1 ms 3596 KiB
testcase15.txt AC 1 ms 3612 KiB
testcase16.txt AC 1 ms 3428 KiB
testcase17.txt AC 18 ms 3632 KiB
testcase18.txt AC 1 ms 3596 KiB
testcase19.txt AC 3 ms 3636 KiB
testcase20.txt AC 1 ms 3452 KiB
testcase21.txt AC 1 ms 3456 KiB
testcase22.txt AC 2 ms 3528 KiB
testcase23.txt AC 8 ms 3588 KiB
testcase24.txt AC 19 ms 3532 KiB
testcase25.txt AC 155 ms 4592 KiB
testcase26.txt AC 183 ms 4828 KiB
testcase27.txt AC 176 ms 4920 KiB
testcase28.txt AC 2 ms 3648 KiB
testcase29.txt AC 3 ms 3664 KiB
testcase30.txt AC 7 ms 3572 KiB
testcase31.txt AC 26 ms 3780 KiB
testcase32.txt AC 197 ms 4868 KiB
testcase33.txt AC 212 ms 5160 KiB
testcase34.txt AC 218 ms 5132 KiB