Submission #2241939


Source Code Expand

#include<bits/stdc++.h>
using namespace std;

#define int long long

#define rep(i,n) for(int i=0;i<(n);i++)
#define pb push_back
#define all(v) (v).begin(),(v).end()
#define fi first
#define se second
typedef vector<int>vint;
typedef pair<int,int>pint;
typedef vector<pint>vpint;

template<typename A,typename B>inline void chmin(A &a,B b){if(a>b)a=b;}
template<typename A,typename B>inline void chmax(A &a,B b){if(a<b)a=b;}

const int mod=1000000007;


int N,K;
int A[222];

signed main(){
    cin>>N>>K;
    rep(i,N)cin>>A[i];
    sort(A,A+N);
    map<vint,vpint>ei;

    for(int i=0;i<=60;i++){
        vint v(N);
        rep(j,N)v[j]=A[j]%(1ll<<i);
        sort(all(v));

        vpint lis;
        if(i<=K)lis.pb({0,K-i});
        for(int j=0;j+1<v.size();j++){
            if(v[j+1]==v[j])continue;
            int cnt=0;
            for(int k=60;k>=0;k--){
                if(v[j+1]>>k&1)cnt++;
                if((v[j+1]>>k&1)!=(v[j]>>k&1))break;
            }
            if(cnt+i>K)continue;
            lis.pb({v[j]+1,K-cnt-i});
        }

        for(auto p:lis){
            vint a(N);
            rep(j,N)a[j]=(A[j]-p.fi)>>i;
            vint d(N-1);
            rep(j,N-1)d[j]=a[j+1]-a[j];
            ei[d].pb({max(0ll,a[0]-p.se),a[0]+1});
        }
    }

    int ans=0;
    for(auto &p:ei){
        vpint v=p.se;

        sort(all(v));
        vpint u;
        for(auto &q:v){
            if(u.size()==0||u.back().se<q.fi)u.pb(q);
            else u.back().se=q.se;
        }
        for(auto &q:u)ans=(ans+q.se-q.fi)%mod;
    }
    cout<<ans<<endl;
    return 0;
}

Submission Info

Submission Time
Task F - Shift and Decrement
User latte0119
Language C++14 (GCC 5.4.1)
Score 1200
Code Size 1656 Byte
Status AC
Exec Time 28 ms
Memory 18176 KB

Judge Result

Set Name Sample All
Score / Max Score 0 / 0 1200 / 1200
Status
AC × 4
AC × 52
Set Name Test Cases
Sample sample1.txt, sample2.txt, sample3.txt, sample4.txt
All sample1.txt, sample2.txt, sample3.txt, sample4.txt, 1.txt, 10.txt, 11.txt, 12.txt, 13.txt, 14.txt, 15.txt, 16.txt, 17.txt, 18.txt, 19.txt, 2.txt, 20.txt, 21.txt, 22.txt, 23.txt, 24.txt, 25.txt, 26.txt, 27.txt, 28.txt, 29.txt, 3.txt, 30.txt, 31.txt, 32.txt, 33.txt, 34.txt, 35.txt, 36.txt, 37.txt, 38.txt, 39.txt, 4.txt, 40.txt, 41.txt, 42.txt, 43.txt, 44.txt, 5.txt, 6.txt, 7.txt, 8.txt, 9.txt, sample1.txt, sample2.txt, sample3.txt, sample4.txt
Case Name Status Exec Time Memory
1.txt AC 28 ms 18176 KB
10.txt AC 26 ms 13952 KB
11.txt AC 26 ms 13824 KB
12.txt AC 28 ms 14848 KB
13.txt AC 27 ms 14336 KB
14.txt AC 9 ms 5376 KB
15.txt AC 5 ms 2432 KB
16.txt AC 28 ms 14976 KB
17.txt AC 27 ms 14336 KB
18.txt AC 26 ms 13952 KB
19.txt AC 26 ms 13824 KB
2.txt AC 28 ms 18176 KB
20.txt AC 28 ms 14976 KB
21.txt AC 25 ms 16256 KB
22.txt AC 17 ms 10240 KB
23.txt AC 28 ms 18176 KB
24.txt AC 28 ms 18176 KB
25.txt AC 28 ms 17920 KB
26.txt AC 27 ms 17792 KB
27.txt AC 26 ms 16640 KB
28.txt AC 27 ms 17536 KB
29.txt AC 25 ms 16256 KB
3.txt AC 28 ms 17920 KB
30.txt AC 17 ms 10240 KB
31.txt AC 27 ms 14336 KB
32.txt AC 27 ms 14080 KB
33.txt AC 27 ms 13952 KB
34.txt AC 28 ms 14848 KB
35.txt AC 27 ms 14336 KB
36.txt AC 5 ms 2560 KB
37.txt AC 27 ms 13824 KB
38.txt AC 28 ms 14976 KB
39.txt AC 27 ms 14208 KB
4.txt AC 28 ms 17792 KB
40.txt AC 27 ms 14080 KB
41.txt AC 27 ms 13824 KB
42.txt AC 28 ms 14976 KB
43.txt AC 26 ms 16256 KB
44.txt AC 17 ms 10112 KB
5.txt AC 27 ms 16512 KB
6.txt AC 27 ms 17408 KB
7.txt AC 26 ms 16256 KB
8.txt AC 17 ms 10240 KB
9.txt AC 27 ms 14336 KB
sample1.txt AC 1 ms 256 KB
sample2.txt AC 1 ms 256 KB
sample3.txt AC 1 ms 256 KB
sample4.txt AC 1 ms 384 KB