結果

問題 No.2918 Divide Applicants Fairly
ユーザー けいと
提出日時 2024-10-09 01:01:43
言語 C
(gcc 13.3.0)
結果
AC  
実行時間 9 ms / 2,000 ms
コード長 3,947 bytes
コンパイル時間 472 ms
コンパイル使用メモリ 31,872 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-10-09 01:01:46
合計ジャッジ時間 2,075 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 61
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <stdio.h>
#include <stdlib.h>
#include <stdint.h>
#include <string.h>
#include <limits.h>

// マクロ定義
#define MAX_N 40
#define MAX_R 800000

// 比較関数（qsort用）
int cmp_uint64(const void *a, const void *b) {
    uint64_t aa = *(const uint64_t*)a;
    uint64_t bb = *(const uint64_t*)b;
    if (aa < bb) return -1;
    if (aa > bb) return 1;
    return 0;
}

// 最小値を返すマクロ
#define MIN(a, b) (((a) < (b)) ? (a) : (b))

int main(){
    // 高速な入出力の設定はCでは不要（scanf/printfは比較的高速）
    
    uint64_t n;
    if (scanf("%llu", &n) != 1) {
        return 1; // 入力エラー
    }
    
    // 配列rの読み込み（n+1サイズに0を追加）
    uint64_t *r = (uint64_t*)malloc((n +1) * sizeof(uint64_t));
    if (r == NULL) {
        return 1; // メモリ確保エラー
    }
    
    for(uint64_t i =0; i <n; ++i){
        if (scanf("%llu", &r[i]) !=1 ){
            free(r);
            return 1; // 入力エラー
        }
    }
    r[n] = 0; // 0を追加
    
    // 配列rをソート
    qsort(r, n, sizeof(uint64_t), cmp_uint64);
    
    // サフィックスサムの計算
    uint64_t *sum = (uint64_t*)malloc((n +1) * sizeof(uint64_t));
    if (sum == NULL){
        free(r);
        return 1; // メモリ確保エラー
    }
    sum[n] = 0;
    for(int i = n -1; i >=0; --i){
        sum[i] = r[i] + sum[i +1];
    }
    
    // mとsize_valの計算
    uint64_t m = n / 2;
    uint64_t size_val = sum[n - m];
    const uint64_t w = 64;
    
    // dp配列のサイズと確保
    uint64_t dp_size = (size_val / w) +1;
    uint64_t *dp = (uint64_t*)calloc(dp_size, sizeof(uint64_t));
    if (dp == NULL){
        free(r);
        free(sum);
        return 1; // メモリ確保エラー
    }
    
    // 初期ビットマスクの設定
    uint64_t bit;
    if(m >= w){
        bit = ~0ULL;
    }
    else{
        bit = (1ULL << m) -1;
    }
    
    // 初期サムの設定
    uint64_t s = sum[0] - sum[m];
    
    // ループの実行
    while(bit < (1ULL << n)){
        uint64_t dp_index = s / w;
        uint64_t dp_bit = s % w;
        
        // dp_indexが範囲内であることを確認
        if(dp_index >= dp_size){
            // サムが範囲外の場合は無視（または適切な処理）
            // ここでは単純に続行
            // 実際の問題に応じて適宜対応してください
        }
        else{
            if( (dp[dp_index] >> dp_bit) & 1 ){
                printf("0\n");
                free(r);
                free(sum);
                free(dp);
                return 0;
            }
            dp[dp_index] |= (1ULL << dp_bit);
        }
        
        if(bit ==0){
            break;
        }
        
        // トレーリングゼロの取得
        int z = __builtin_ctzll(bit);
        
        // トレーリングワンのカウント
        uint64_t shifted = bit >> z;
        int o = 0;
        while(shifted &1){
            o++;
            shifted >>=1;
        }
        
        // サムの更新
        s += r[z + o];
        s -= (sum[z] - sum[z + o]);
        s += (sum[0] - sum[o -1]);
        
        // ビットマスクの更新
        bit = (bit + (1ULL << z)) | ((1ULL << (o -1)) -1);
    }
    
    // 最小差分の計算
    uint64_t ans = ULLONG_MAX;
    uint64_t up = ULLONG_MAX;
    
    for(uint64_t i = dp_size; i-- >0; ){
        uint64_t current = dp[i];
        while(current >0){
            // 最上位ビットの位置を取得
            int pos = 63 - __builtin_clzll(current);
            uint64_t u = i * w + pos;
            if(up != ULLONG_MAX){
                ans = MIN(ans, up - u);
            }
            up = u;
            current &= ~(1ULL << pos);
        }
    }
    
    printf("%llu\n", ans);
    
    // メモリの解放
    free(r);
    free(sum);
    free(dp);
    
    return 0;
}
0