CSES - Aalto Competitive Programming 2024 - wk9 - Mon - Results
Submission details
Task:Matter++
Sender:AleksandrPolitov
Submission time:2024-11-04 17:30:26 +0200
Language:C++ (C++20)
Status:READY
Result:ACCEPTED
Test results
testverdicttime
#1ACCEPTED0.00 sdetails
#2ACCEPTED0.00 sdetails
#3ACCEPTED0.00 sdetails
#4ACCEPTED0.00 sdetails
#5ACCEPTED0.00 sdetails
#6ACCEPTED0.00 sdetails
#7ACCEPTED0.00 sdetails
#8ACCEPTED0.00 sdetails
#9ACCEPTED0.01 sdetails
#10ACCEPTED0.00 sdetails
#11ACCEPTED0.00 sdetails
#12ACCEPTED0.00 sdetails
#13ACCEPTED0.00 sdetails
#14ACCEPTED0.00 sdetails
#15ACCEPTED0.00 sdetails
#16ACCEPTED0.00 sdetails
#17ACCEPTED0.00 sdetails
#18ACCEPTED0.00 sdetails
#19ACCEPTED0.00 sdetails
#20ACCEPTED0.00 sdetails
#21ACCEPTED0.00 sdetails
#22ACCEPTED0.00 sdetails
#23ACCEPTED0.00 sdetails
#24ACCEPTED0.00 sdetails
#25ACCEPTED0.00 sdetails
#26ACCEPTED0.00 sdetails
#27ACCEPTED0.00 sdetails
#28ACCEPTED0.00 sdetails
#29ACCEPTED0.00 sdetails
#30ACCEPTED0.00 sdetails
#31ACCEPTED0.00 sdetails
#32ACCEPTED0.00 sdetails
#33ACCEPTED0.00 sdetails
#34ACCEPTED0.00 sdetails
#35ACCEPTED0.00 sdetails
#36ACCEPTED0.00 sdetails
#37ACCEPTED0.00 sdetails
#38ACCEPTED0.00 sdetails
#39ACCEPTED0.00 sdetails
#40ACCEPTED0.00 sdetails
#41ACCEPTED0.00 sdetails
#42ACCEPTED0.00 sdetails
#43ACCEPTED0.00 sdetails
#44ACCEPTED0.00 sdetails
#45ACCEPTED0.00 sdetails
#46ACCEPTED0.00 sdetails
#47ACCEPTED0.00 sdetails
#48ACCEPTED0.00 sdetails
#49ACCEPTED0.00 sdetails
#50ACCEPTED0.00 sdetails
#51ACCEPTED0.00 sdetails
#52ACCEPTED0.01 sdetails
#53ACCEPTED0.01 sdetails
#54ACCEPTED0.01 sdetails
#55ACCEPTED0.01 sdetails
#56ACCEPTED0.01 sdetails
#57ACCEPTED0.01 sdetails
#58ACCEPTED0.01 sdetails
#59ACCEPTED0.01 sdetails
#60ACCEPTED0.01 sdetails
#61ACCEPTED0.01 sdetails
#62ACCEPTED0.16 sdetails
#63ACCEPTED0.16 sdetails
#64ACCEPTED0.17 sdetails
#65ACCEPTED0.17 sdetails
#66ACCEPTED0.17 sdetails
#67ACCEPTED0.16 sdetails
#68ACCEPTED0.16 sdetails
#69ACCEPTED0.17 sdetails
#70ACCEPTED0.16 sdetails
#71ACCEPTED0.17 sdetails

Code

#ifdef ONPC
    #define _GLIBCXX_DEBUG
#endif
#include <bits/stdc++.h>

#define char unsigned char
#define rep(i, a, b) for(int i=a; i< (b); ++i)
#define all(x) begin(x), end(x)
#define sz(x) (int)(x).size()
#define eb emplace_back
#define mp make_pair
#define mt make_tuple
#define fi first
#define se second
#define pb push_back

#define LSOne(S) ((S) & -(S))

using namespace std;
// mt19937 rnd(239);
mt19937 rnd(chrono::steady_clock::now().time_since_epoch().count());

template <typename T> int sgn(T x) { return (T(0) < x) - (x < T(0)); }
typedef long double T;
typedef complex<T> pt;
#define X real()
#define Y imag()

template<class T>
istream& operator>> (istream& is, complex<T>& p) {
    T value;
    is >> value;
    p.real(value);
    is >> value;
    p.imag(value);
    return is;
}

typedef long long ll;
typedef long double ld;

using pi = pair<ll, ll>;
using vi = vector<ll>;
template <class T>
using pq = priority_queue<T>;
template <class T>
using pqg = priority_queue<T, vector<T>, greater<T>>;

int popcnt(int x) { return __builtin_popcount(x); }
int popcnt(ll x) { return __builtin_popcountll(x); }

#define MIN(v) *min_element(all(v))
#define MAX(v) *max_element(all(v))
#define LB(c, x) distance((c).begin(), lower_bound(all(c), (x)))
#define UB(c, x) distance((c).begin(), upper_bound(all(c), (x)))

void __print(int x) {cerr << x;}
void __print(long x) {cerr << x;}
void __print(long long x) {cerr << x;}
void __print(unsigned x) {cerr << x;}
void __print(unsigned long x) {cerr << x;}
void __print(unsigned long long x) {cerr << x;}
void __print(float x) {cerr << x;}
void __print(double x) {cerr << x;}
void __print(long double x) {cerr << x;}
void __print(char x) {cerr << '\'' << x << '\'';}
void __print(const char *x) {cerr << '\"' << x << '\"';}
void __print(const string &x) {cerr << '\"' << x << '\"';}
void __print(bool x) {cerr << (x ? "true" : "false");}

template<typename T, typename V>
void __print(const pair<T, V> &x) {cerr << '{'; __print(x.first); cerr << ", "; __print(x.second); cerr << '}';}
template<typename T>
void __print(const T &x) {int f = 0; cerr << '{'; for (auto &i: x) cerr << (f++ ? ", " : ""), __print(i); cerr << "}";}
void _print() {cerr << "]\n";}
template <typename T, typename... V>
void _print(T t, V... v) {__print(t); if (sizeof...(v)) cerr << ", "; _print(v...);}
#ifdef DEBUG
#define dbg(x...) cerr << "\e[91m"<<__func__<<":"<<__LINE__<<" [" << #x << "] = ["; _print(x); cerr << "\e[39m" << endl;
#else
#define dbg(x...)
#endif

template<typename S, typename T = S> void chmin(S &s, T t) {s = s < t ? s : t;}
template<typename S, typename T = S> void chmax(S &s, T t) {s = s > t ? s : t;}

const int INF = 1e9; // 10^9 = 1B is < 2^31-1
const ll LLINF = 4e18; // 4*10^18 is < 2^63-1
const double EPS = 1e-9;
const ll MOD = 1e9+7;

const int N=1e5+10;
ll pref[N][26];
ll W;

int n;

ll calc(int i, int j) {
    j=min(n-1, j);
    if(i>j) return 0;
    vector<int> curr(26, 0);
    for (int q = 0; q < 26; q++) {
        curr[q]=pref[j+1][q]-pref[i][q];
    }

    if(false) {
    std::cout << i << "-" << j  << std::endl;
    for (int q = 0; q < 26; q++) {
        std::cout << curr[q] << " ";
    }
    std::cout  << std::endl;
    }
    ll res=0;
    for (int q = 0; q < 26; q++) {
        res+=(curr[q]*curr[q]);
    }

    return res;
}

bool ok(int i, int j) {
     return calc(i, j)>=W;
}



int solve() {
    std::cin >> W;
    string s; cin >> s;
    n=s.size();

    for (int i = 0; i < n; i++) {
        for (int j = 0; j < 26; j++) pref[i+1][j]=pref[i][j];
        pref[i+1][s[i]-'a']++;
    }

    if(false)
    for (int i = 0; i < n; i++) {
        for (int j = 0; j < 26; j++) {
            std::cout << pref[i][j] << " ";
        }
        std::cout  << std::endl;
    }

    //calc(2,n-1);
    //return 0;


    ll tmp=0;
    for (int i = 0; i < n; i++) {
        tmp+=(pref[n][i]-pref[0][i])*(pref[n][i]-pref[0][i]);
    }


    //std::cout << "W: " << W << std::endl;
    

    pair<ll, int> ans={n, 0};
    for (int i = 0; i < n; i++) {
        int lo=i-1, hi=n-1;
        while(lo<hi) {
            int mid=(lo+hi+1)/2;
            if(ok(i, mid)) {
                hi=mid-1;
            } else {
                lo=mid;
            }
        }

        //std::cout << i << " " << lo+1 << ", " << calc(i, lo+1) << ": " << s.substr(i, lo+1-i+1) << std::endl;
        if(lo+1-i+1<ans.fi && calc(i, lo+1)>=W) {
            ans={lo+1-i+1, i};
            //std::cout << calc(i, lo+1) << std::endl;
            //std::cout << "new ans: " << s.substr(ans.se, ans.fi) << std::endl;
        }
    }
    //__print(ans); std::cout  << std::endl;

    std::cout << s.substr(ans.se, ans.fi) << std::endl;
    

    
    return 0;
}

int32_t main() {
    ios::sync_with_stdio(0);
    cin.tie(0);
    int TET = 1;
    //cin >> TET;
    for (int i = 1; i <= TET; i++) {
        #ifdef ONPC
            cout << "TEST CASE#" << i << endl;
        #endif
        
        if (solve()) {
            break;
        }

        #ifdef ONPC
            cout << "__________________________" << endl;
        #endif
    }
    #ifdef ONPC
        cerr << endl << "finished in " << clock() * 1.0 / CLOCKS_PER_SEC << " sec" << endl;
    #endif
}

Test details

Test 1

Verdict: ACCEPTED

input
1
e

correct output
e

user output
e

Test 2

Verdict: ACCEPTED

input
1
be

correct output
b

user output
b

Test 3

Verdict: ACCEPTED

input
1
za

correct output
z

user output
z

Test 4

Verdict: ACCEPTED

input
1
po

correct output
p

user output
p

Test 5

Verdict: ACCEPTED

input
1
acc

correct output
a

user output
a

Test 6

Verdict: ACCEPTED

input
2
caa

correct output
ca

user output
ca

Test 7

Verdict: ACCEPTED

input
2
caa

correct output
ca

user output
ca

Test 8

Verdict: ACCEPTED

input
2
cced

correct output
cc

user output
cc

Test 9

Verdict: ACCEPTED

input
2
ceea

correct output
ce

user output
ce

Test 10

Verdict: ACCEPTED

input
1
ccad

correct output
c

user output
c

Test 11

Verdict: ACCEPTED

input
4
ccda

correct output
cc

user output
cc

Test 12

Verdict: ACCEPTED

input
1
defdf

correct output
d

user output
d

Test 13

Verdict: ACCEPTED

input
1
fefaa

correct output
f

user output
f

Test 14

Verdict: ACCEPTED

input
7
bafdf

correct output
bafdf

user output
bafdf

Test 15

Verdict: ACCEPTED

input
1
aefba

correct output
a

user output
a

Test 16

Verdict: ACCEPTED

input
3
cbacc

correct output
cc

user output
cc

Test 17

Verdict: ACCEPTED

input
1
bwvfj

correct output
b

user output
b

Test 18

Verdict: ACCEPTED

input
11
caaca

correct output
caaca

user output
caaca

Test 19

Verdict: ACCEPTED

input
4
fuilz

correct output
fuil

user output
fuil

Test 20

Verdict: ACCEPTED

input
5
acacb

correct output
aca

user output
aca

Test 21

Verdict: ACCEPTED

input
4
jnmma

correct output
mm

user output
mm

Test 22

Verdict: ACCEPTED

input
6
defdfdfcdd

correct output
defd

user output
defd

Test 23

Verdict: ACCEPTED

input
5
fefaabfaba

correct output
fef

user output
fef

Test 24

Verdict: ACCEPTED

input
22
bafdfcccbb

correct output
bafdfcccbb

user output
bafdfcccbb

Test 25

Verdict: ACCEPTED

input
12
aefbaddfcf

correct output
aefbaddf

user output
aefbaddf

Test 26

Verdict: ACCEPTED

input
14
cbacccbcba

correct output
cccbc

user output
cccbc

Test 27

Verdict: ACCEPTED

input
1
bwvfjxzmcp

correct output
b

user output
b

Test 28

Verdict: ACCEPTED

input
17
caacaabacb

correct output
aacaa

user output
aacaa

Test 29

Verdict: ACCEPTED

input
13
fuilzslzio

correct output
ilzslzi

user output
ilzslzi

Test 30

Verdict: ACCEPTED

input
16
acacbbcaba

correct output
acacbbc

user output
acacbbc

Test 31

Verdict: ACCEPTED

input
5
jnmmadidaf

correct output
nmm

user output
nmm

Test 32

Verdict: ACCEPTED

input
278
defdfdfcddccbfafbcceedcdcffaca...

correct output
fdfcddccbfafbcceedcdcffacadace...

user output
fdfcddccbfafbcceedcdcffacadace...

Test 33

Verdict: ACCEPTED

input
32
fefaabfabacbccecfdfcbedbcfbade...

correct output
cbccecfdfc

user output
cbccecfdfc

Test 34

Verdict: ACCEPTED

input
1387
bafdfcccbbabedabcbddedeacddbee...

correct output
dfcccbbabedabcbddedeacddbeebfd...

user output
dfcccbbabedabcbddedeacddbeebfd...

Test 35

Verdict: ACCEPTED

input
190
aefbaddfcfaaabbaaceaaccdbbcecd...

correct output
aaabbaaceaaccdbbcecdfaedfbccdb...

user output
aaabbaaceaaccdbbcecdfaedfbccdb...

Test 36

Verdict: ACCEPTED

input
1205
cbacccbcbaacaacacbbcaabcbcbabb...

correct output
abbaabccbacacccbabaabaacccbcbc...

user output
abbaabccbacacccbabaabaacccbcbc...

Test 37

Verdict: ACCEPTED

input
299
bwvfjxzmcpktjnmhzevcqtvlgetwch...

correct output
vfjxzmcpktjnmhzevcqtvlgetwchfk...

user output
vfjxzmcpktjnmhzevcqtvlgetwchfk...

Test 38

Verdict: ACCEPTED

input
2796
caacaabacbabbbbbbbcbccbbababcc...

correct output
bacbabbbbbbbcbccbbababccccbbcc...

user output
bacbabbbbbbbcbccbbababccccbbcc...

Test 39

Verdict: ACCEPTED

input
345
fuilzslziogncbkgamnrwuijkbrhrx...

correct output
lziogncbkgamnrwuijkbrhrxsfilyy...

user output
lziogncbkgamnrwuijkbrhrxsfilyy...

Test 40

Verdict: ACCEPTED

input
1517
acacbbcababbcbbbcbcbabccbccbab...

correct output
baaaaaccaaaaababbccbbccaaababb...

user output
baaaaaccaaaaababbccbbccaaababb...

Test 41

Verdict: ACCEPTED

input
123
jnmmadidafwkdgdckimeiwjytbzsso...

correct output
mmadidafwkdgdckimeiwjytbzssodx...

user output
mmadidafwkdgdckimeiwjytbzssodx...

Test 42

Verdict: ACCEPTED

input
1079
defdfdfcddccbfafbcceedcdcffaca...

correct output
fdfdfcddccbfafbcceedcdcffacada...

user output
fdfdfcddccbfafbcceedcdcffacada...

Test 43

Verdict: ACCEPTED

input
56
fefaabfabacbccecfdfcbedbcfbade...

correct output
fbeeefebaedfff

user output
fbeeefebaedfff

Test 44

Verdict: ACCEPTED

input
5337
bafdfcccbbabedabcbddedeacddbee...

correct output
bafdfcccbbabedabcbddedeacddbee...

user output
bafdfcccbbabedabcbddedeacddbee...
Truncated

Test 45

Verdict: ACCEPTED

input
744
aefbaddfcfaaabbaaceaaccdbbcecd...

correct output
cccacdedecabbcdcebdbbcdfddfbeb...

user output
cccacdedecabbcdcebdbbcdfddfbeb...

Test 46

Verdict: ACCEPTED

input
10488
cbacccbcbaacaacacbbcaabcbcbabb...

correct output
bcbcbabbcabbaabccbacacccbabaab...

user output
bcbcbabbcabbaabccbacacccbabaab...
Truncated

Test 47

Verdict: ACCEPTED

input
11
bwvfjxzmcpktjnmhzevcqtvlgetwch...

correct output
pxyyy

user output
pxyyy

Test 48

Verdict: ACCEPTED

input
5929
caacaabacbabbbbbbbcbccbbababcc...

correct output
bbbbbbbcbccbbababccccbbccccbac...

user output
bbbbbbbcbccbbababccccbbccccbac...
Truncated

Test 49

Verdict: ACCEPTED

input
1398
fuilzslziogncbkgamnrwuijkbrhrx...

correct output
gncbkgamnrwuijkbrhrxsfilyygatp...

user output
gncbkgamnrwuijkbrhrxsfilyygatp...
Truncated

Test 50

Verdict: ACCEPTED

input
5219
acacbbcababbcbbbcbcbabccbccbab...

correct output
acbbcababbcbbbcbcbabccbccbabba...

user output
acbbcababbcbbbcbcbabccbccbabba...
Truncated

Test 51

Verdict: ACCEPTED

input
450
jnmmadidafwkdgdckimeiwjytbzsso...

correct output
qmtyjuvzcecnbymqfwyzzergpwuxbt...

user output
qmtyjuvzcecnbymqfwyzzergpwuxbt...
Truncated

Test 52

Verdict: ACCEPTED

input
26679
defdfdfcddccbfafbcceedcdcffaca...

correct output
bcbfacacffbedecaafcecffaffaadd...

user output
bcbfacacffbedecaafcecffaffaadd...
Truncated

Test 53

Verdict: ACCEPTED

input
22417
fefaabfabacbccecfdfcbedbcfbade...

correct output
adefccdcafbeeefebaedfffeaeabba...

user output
adefccdcafbeeefebaedfffeaeabba...
Truncated

Test 54

Verdict: ACCEPTED

input
128464
bafdfcccbbabedabcbddedeacddbee...

correct output
bafdfcccbbabedabcbddedeacddbee...

user output
bafdfcccbbabedabcbddedeacddbee...
Truncated

Test 55

Verdict: ACCEPTED

input
10769
aefbaddfcfaaabbaaceaaccdbbcecd...

correct output
ebefdceecdabceccdeaebdeebdcdfa...

user output
ebefdceecdabceccdeaebdeebdcdfa...
Truncated

Test 56

Verdict: ACCEPTED

input
70653
cbacccbcbaacaacacbbcaabcbcbabb...

correct output
acaaaaaabaaabbccbabbbbcacbccbc...

user output
acaaaaaabaaabbccbabbbbcacbccbc...
Truncated

Test 57

Verdict: ACCEPTED

input
233
bwvfjxzmcpktjnmhzevcqtvlgetwch...

correct output
qfvyojqjvhyfxtvycrjlaoofuebvev...

user output
qfvyojqjvhyfxtvycrjlaoofuebvev...

Test 58

Verdict: ACCEPTED

input
116525
caacaabacbabbbbbbbcbccbbababcc...

correct output
caaabbacabbbabcccbacaaacaaccac...

user output
caaabbacabbbabcccbacaaacaaccac...
Truncated

Test 59

Verdict: ACCEPTED

input
27985
fuilzslziogncbkgamnrwuijkbrhrx...

correct output
uilzslziogncbkgamnrwuijkbrhrxs...

user output
uilzslziogncbkgamnrwuijkbrhrxs...
Truncated

Test 60

Verdict: ACCEPTED

input
161959
acacbbcababbcbbbcbcbabccbccbab...

correct output
acacbbcababbcbbbcbcbabccbccbab...

user output
acacbbcababbcbbbcbcbabccbccbab...
Truncated

Test 61

Verdict: ACCEPTED

input
8903
jnmmadidafwkdgdckimeiwjytbzsso...

correct output
uaguxmabjxfuialfmskifofagvsnkn...

user output
uaguxmabjxfuialfmskifofagvsnkn...
Truncated

Test 62

Verdict: ACCEPTED

input
264240357
defdfdfcddccbfafbcceedcdcffaca...

correct output
badeaccfbabfcceaabffecbffcdcfe...

user output
badeaccfbabfcceaabffecbffcdcfe...
Truncated

Test 63

Verdict: ACCEPTED

input
30437082
fefaabfabacbccecfdfcbedbcfbade...

correct output
acdbbfafacdebdefccfbcfcdeecdfd...

user output
acdbbfafacdebdefccfbcfcdeecdfd...
Truncated

Test 64

Verdict: ACCEPTED

input
1280505429
bafdfcccbbabedabcbddedeacddbee...

correct output
eaceacebbcdfacddcecfceccddfcac...

user output
eaceacebbcdfacddcecfceccddfcac...
Truncated

Test 65

Verdict: ACCEPTED

input
107609709
aefbaddfcfaaabbaaceaaccdbbcecd...

correct output
bffcffbddbfedcccfcdcdfecbbefbe...

user output
bffcffbddbfedcccfcdcdfecbbefbe...
Truncated

Test 66

Verdict: ACCEPTED

input
706196896
cbacccbcbaacaacacbbcaabcbcbabb...

correct output
bccbccccacabcacaaabcccbacbcaba...

user output
bccbccccacabcacaaabcccbacbcaba...
Truncated

Test 67

Verdict: ACCEPTED

input
9966992
bwvfjxzmcpktjnmhzevcqtvlgetwch...

correct output
hljwtcgatnetpvfxeiipnhaarsejbf...

user output
hljwtcgatnetpvfxeiipnhaarsejbf...
Truncated

Test 68

Verdict: ACCEPTED

input
1449002782
caacaabacbabbbbbbbcbccbbababcc...

correct output
acbaabcaababcabcccabcbbaacccca...

user output
acbaabcaababcabcccabcbbaacccca...
Truncated

Test 69

Verdict: ACCEPTED

input
275504573
fuilzslziogncbkgamnrwuijkbrhrx...

correct output
lziogncbkgamnrwuijkbrhrxsfilyy...

user output
lziogncbkgamnrwuijkbrhrxsfilyy...
Truncated

Test 70

Verdict: ACCEPTED

input
1480920774
acacbbcababbcbbbcbcbabccbccbab...

correct output
bcccaccabcaabccaaabcbbacbbbbba...

user output
bcccaccabcaabccaaabcbbacbbbbba...
Truncated

Test 71

Verdict: ACCEPTED

input
199518168
jnmmadidafwkdgdckimeiwjytbzsso...

correct output
mzdiyihtesgshxbshryvzqghavzgrs...

user output
mzdiyihtesgshxbshryvzqghavzgrs...
Truncated