#391132 (C++14) No.911 ラッキーソート

• 問題: No.911 ラッキーソート
• ユーザー: polylogK
• 提出日時: 2019-10-18 22:15:24
• 言語: C++14 (gcc 13.3.0 + boost 1.87.0)
• 結果: AC
• 実行時間: 62 ms / 2,000 ms
• コード長: 2,108 bytes
• コンパイル時間: 1,488 ms
• コンパイル使用メモリ: 173,444 KB
• 実行使用メモリ: 5,760 KB
• 最終ジャッジ日時: 2024-06-25 17:11:03
• 合計ジャッジ時間: 4,621 ms
• ジャッジサーバーID （参考情報）: judge4 / judge2

ファイルパターン	結果
sample	AC * 3
other	AC * 46

ソースコード

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
#include <bits/stdc++.h>
using namespace std::literals::string_literals;
using i64 = std::int_fast64_t;
using std::cout;
using std::endl;
using std::cin;
template<typename T>
std::vector<T> make_v(size_t a){return std::vector<T>(a);}
template<typename T,typename... Ts>
auto make_v(size_t a,Ts... ts){
  return std::vector<decltype(make_v<T>(ts...))>(a,make_v<T>(ts...));
}
int main() {
    int n; i64 L, R; scanf("%d%lld%lld", &n, &L, &R);
    std::vector<i64> a(n);
    for(int i = 0; i < n; i++) scanf("%lld", &a[i]);
    // 0011 | 0011 | 0011 -> 1 pattern
    // 0011 | 1100 | .... -> inpossiblent i = 0; i < 60; i++) {
    // 0000000 or 1111111 -> 2 pattern
    std::vector<int> kiri(n + 1, false); kiri[n] = true; // 数列切るたん, left)
    std::vector<int> tan(61, 0); // true -> 1 pattern
    for(int k = 60; k >= 0; k--) {
        // x: 1100
        // y: 0011
        // z: 0000 or 1111
        int x = 0, y = 0, z = 0, w = 0;
        for(int i = 0; i < n; i++) {
            if(kiri[i + 1]) {
                if(w > 1) {
                    printf("0\n");
                    return 0;
                } else if(w == 0) {
                    z++;
                }
                w = 0;
                
                continue;
            }
            
            int A = a[i] >> k & 1, B = a[i + 1] >> k & 1;
            if(A == B) continue;
            kiri[i + 1] = true;
            if(A) x++; 
            else y++;
            w++;
        }
        if(x and y) {
            printf("0\n");
            return 0;
        }
        if(x) tan[k] = 1;
        else if(y) tan[k] = 0;
        else tan[k] = -1;
    }
    
    auto calc = [&](i64 t) -> i64 {
        if(t < 0) return 0LL;
    
        auto dp = make_v<i64>(62, 2); dp[0][1] = 1;
        for(int k = 60, i = 0; k >= 0; k--, i++) {
            if(t >> k & 1) {
                if(tan[k] == 0) {
                    dp[i + 1][0] += dp[i][0] + dp[i][1];
                } else  if(tan[k] == 1) {
                    dp[i + 1][0] += dp[i][0];
                    dp[i + 1][1] += dp[i][1];
                } else {
                    dp[i + 1][0] += dp[i][0] * 2LL + dp[i][1];
                    dp[i + 1][1] += dp[i][1];
                }
            } else {
                if(tan[k] == 0) {
                    dp[i + 1][0] += dp[i][0];
                    dp[i + 1][1] += dp[i][1];
                } else if(tan[k] == 1) {
                    dp[i + 1][0] += dp[i][0];
                } else {
                    dp[i + 1][0] += dp[i][0] * 2LL;
                    dp[i + 1][1] += dp[i][1];
                }
            }
        }
        return (dp[61][0] + dp[61][1]);
    };
    
    printf("%lld\n", calc(R) - calc(L - 1));
    return 0;
}
 
 
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX