Submission #265463

#	Time	Username	Problem	Language	Result	Execution time	Memory
265463		fivefourthreeone	Sequence (BOI14_sequence)	C++17	0 / 100	205 ms	1248 KiB

sequence

//#pragma GCC target ("avx2")
#pragma GCC optimization ("O3")
#pragma GCC optimization ("unroll-loops")
#include <bits/stdc++.h>
#define owo(i,a, b) for(int i=(a);i<(b); ++i)
#define uwu(i,a, b) for(int i=(a)-1; i>=(b); --i)
#define senpai push_back
#define ttgl pair<int, int>
#define ayaya cout<<"ayaya~"<<endl
using namespace std;
/*#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
using namespace __gnu_pbds;
#define ordered_set tree<ttgl, null_type,less<ttgl>, rb_tree_tag,tree_order_statistics_node_update>*/
 
using ll = long long;
using ld = long double;
const ll MOD = 1000000007;
const ll root = 62;
int gcd(int a,int b){return b?gcd(b,a%b):a;}
ll binpow(ll a,ll b){ll res=1;while(b){if(b&1)res=(res*a)%MOD;a=(a*a)%MOD;b>>=1;}return res;}
ll modInv(ll a){return binpow(a, MOD-2);}
const double PI = acos(-1);
const double eps = 1e-6;
const int INF = 0x3f3f3f3f;
const int NINF = 0xc0c0c0c0;
const ll INFLL = 0x3f3f3f3f3f3f3f3f;
const ll NINFLL = 0xc0c0c0c0c0c0c0c0;
const int mxN = 100001;
ll p10[19];
int arr[mxN];
ll ans = INFLL;
int k;
//after dealing with one digit, we are dealing with n/10 digits the next iteration
void solve(vector<int> all, ll num, int dig) {
    if(num>ans)return;
    if(all.size()==1) {
        if(all[0]==1) {
            num+=p10[dig+1];
        }else {
            if(all[0]&1) {
                int k = 0;
                uwu(i, 9, 1) {
                    if(all[0]&(1<<i))k = i;
                }
                uwu(i, 9, k+1) {
                    if(all[0]&(1<<i))num+=1LL*i*p10[dig++];
                }
                dig++;
                num+=1LL*k*p10[dig];
            }else {
                uwu(i, 9, 1) {
                    if(all[0]&(1<<i))num+=1LL*i*p10[dig++];
                }
            }
        }
        ans = min(ans, num);
        return;
    }
    owo(i, 0, 10) {
        vector<int> nxt;
        int curr= i;
        int msk = 0;
        int _ = (1<<10)-1;
        owo(j, 0, all.size()) {
            msk|=(all[j]&(_^(1<<curr)));
            curr++;
            if(curr==10) {
                nxt.senpai(msk);
                msk = curr = 0;
            }
        }
        if(curr) {
            nxt.senpai(msk);
        }
        solve(nxt, num+i*p10[dig], dig+1);
    }
}
int main() {
    //freopen("file.in", "r", stdin);
    //freopen("file.out", "w", stdout);
    mt19937_64 rng(chrono::steady_clock::now().time_since_epoch().count());
    cin.tie(0)->sync_with_stdio(0);
    cin>>k;
    owo(i, 0, k) {
        cin>>arr[i];
    }
    vector<int> val;
    p10[0] = 1;
    owo(i, 1, 19) {
        p10[i] = p10[i-1]*10;
    }
    owo(i, 0, k) {
        val.senpai(1<<arr[i]);
    }
    solve(val, 0, 0);
    cout<<ans<<"\n";
    return 0;
}

Compilation message (stderr)

sequence.cpp:2: warning: ignoring #pragma GCC optimization [-Wunknown-pragmas]
    2 | #pragma GCC optimization ("O3")
      | 
sequence.cpp:3: warning: ignoring #pragma GCC optimization [-Wunknown-pragmas]
    3 | #pragma GCC optimization ("unroll-loops")
      | 
sequence.cpp: In function 'void solve(std::vector<int>, ll, int)':
sequence.cpp:5:36: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
    5 | #define owo(i,a, b) for(int i=(a);i<(b); ++i)
      |                                    ^
sequence.cpp:65:9: note: in expansion of macro 'owo'
   65 |         owo(j, 0, all.size()) {
      |         ^~~

• Subtask 1: 0 / 9
• Subtask 2: 0 / 33
• Subtask 3: 0 / 25
• Subtask 4: 0 / 33