Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- using namespace std;
- int main() {
- ios_base::sync_with_stdio(false);
- cin.tie(0);
- cout.tie(0);
- int n, k;
- cin >> n >> k;
- vector<int> a(n);
- vector<int> ans(n);
- for(int i = 0; i < n; ++i){
- cin >> a[i];
- }
- int x = 0;
- int y = 0;
- int res = k - 1;
- for(int i = 0; i < n; ++i){
- if(a[i] == 1){
- ++x;
- }
- else{
- ++y;
- }
- if(min(x, y) + (k - i - 1) < n || max(x, y) == n){
- res = i;
- break;
- }
- }
- cout << res + 1;
- cout << endl;
- for(int i = 0; i <= min(k - 1 ,res); ++i){
- cout << 0 << " ";
- }
- for(int i = res + 1; i < k; ++i){
- cout << 1 << " ";
- }
- cout << endl;
- }
Advertisement
Add Comment
Please, Sign In to add comment