Submission #1152162

#TimeUsernameProblemLanguageResultExecution timeMemory
1152162dostsDetecting Molecules (IOI16_molecules)C++17
Compilation error
0 ms0 KiB
#include "molecules.h"
#include <bits/stdc++.h>
# define int long long
#pragma GCC optimize("O3")
#define vi vector<int>
#define vb vector<bool>
#define F(i, s, e) for(int i = s; i < e; i++)
#define sp <<' '<<
#define pii pair<int, int>
#define fi first
#define se second
#define pb push_back
#define vvi vector<vi>
#define vpi vector<pii>
#define vvpi vector<vpi>
#define endl '\n'
 
const int mod = 1e9 + 7;
 
using namespace std;
 
const int N = 2e5+5;
const int inf = 1e12;

vector<int32_t> find_subset(int32_t L, int32_t R, vector<int32_t> b) {
    int n = b.size();
    vpi a;
    F(i, 0, n) a.pb({b[i], i});

    sort(a.begin(), a.end());

    vi pref(n, 0);
    vi suff(n, 0);
    pref[0] = a[0].fi;
    F(i, 1, n) pref[i] = pref[i-1] + a[i].fi;
    suff[n-1] = a[n-1].fi;
    for(int i = n-2; i >= 0; i--) suff[i] = suff[i+1] + a[i].fi;

    F(i, 0, n) {
        if(pref[i] >= L && pref[i] <= R) {
            vi res;
            F(j, 0, i+1) res.pb(a[j].se);
            return res;
        }

        if(suff[n-1-i] >= L && suff[n-1-i] <= R) {
            vi res;
            F(j, n-1-i, n) res.pb(a[j].se);
            return res;
        }

        if(pref[i] < L && suff[n-1-i] > R) {
            // ans is somewhere in the middle
            vi res;
            F(j, 0, i+1) res.pb(a[j].se);
            int summy = pref[i];

            F(j, 0, i+1) {
                res[j] = a[n-1-j].se;
                summy -= a[j].fi;
                summy += a[n-1-j].fi;

                if(summy >= L && summy <= R) return res;
            }
        }
    }
    return vi(0);
}

Compilation message (stderr)

molecules.cpp: In function 'std::vector<int> find_subset(int32_t, int32_t, std::vector<int>)':
molecules.cpp:43:20: error: could not convert 'res' from 'vector<long long int>' to 'vector<int>'
   43 |             return res;
      |                    ^~~
      |                    |
      |                    vector<long long int>
molecules.cpp:49:20: error: could not convert 'res' from 'vector<long long int>' to 'vector<int>'
   49 |             return res;
      |                    ^~~
      |                    |
      |                    vector<long long int>
molecules.cpp:63:53: error: could not convert 'res' from 'vector<long long int>' to 'vector<int>'
   63 |                 if(summy >= L && summy <= R) return res;
      |                                                     ^~~
      |                                                     |
      |                                                     vector<long long int>
molecules.cpp:5:12: error: could not convert 'std::vector<long long int>(0, std::allocator<long long int>())' from 'vector<long long int>' to 'vector<int>'
    5 | #define vi vector<int>
      |            ^
      |            |
      |            vector<long long int>
molecules.cpp:67:12: note: in expansion of macro 'vi'
   67 |     return vi(0);
      |            ^~
molecules.h:1:9: warning: #pragma once in main file
    1 | #pragma once
      |         ^~~~
molecules_c.h:1:9: warning: #pragma once in main file
    1 | #pragma once
      |         ^~~~