Submission #1198774

#	Time	Username	Problem	Language	Result	Execution time	Memory
1198774		lrnnz	Paint By Numbers (IOI16_paint)	C++20	32 / 100	0 ms	328 KiB

paint

#include <bits/stdc++.h>
#include <iostream>
#include <vector>
#include <algorithm>
#include <cmath>
#include <iomanip>
#include "paint.h"
using namespace std;
 
#define all(a) (a).begin(), (a).end()
#define sz(a) (int)(a).size()
#define ll long long
#define ld long double
#define ui uint64_t
#define cont(set, element) ((set).find(element) != (set).end())
 
/********* DEBUG *********/
 
template <typename T>
void outvec(const vector<T>& Z){
    for (const T& x : Z)
    cout << x << ' ';
    cout << "\n";
}
void printVariable(const any& var) {
    if (!var.has_value()) {
        cout << "null";
        return;
    }

    if (var.type() == typeid(int)) {
        cout << any_cast<int>(var);
    } else if (var.type() == typeid(double)) {
        cout << any_cast<double>(var);
    } else if (var.type() == typeid(float)) {
        cout << any_cast<float>(var);
    } else if (var.type() == typeid(char)) {
        cout << any_cast<char>(var);
    } else if (var.type() == typeid(bool)) {
        cout << (any_cast<bool>(var) ? "true" : "false");
    } else if (var.type() == typeid(string)) {
        cout << any_cast<string>(var);
    } else if (var.type() == typeid(const char*)) {
        cout << any_cast<const char*>(var);
    } else if (var.type() == typeid(long long)) {
        cout << any_cast<long long>(var);
    } else {
        cout << "[unknown type]";
    }
}

template<typename... Args>
void outval(Args... args) {
    vector<any> variables = {args...};
    
    for (size_t i = 0; i < variables.size(); ++i) {
        printVariable(variables[i]);
        if (i != variables.size() - 1) {
            cout << " ";
        }
    }
    cout << "\n";
}

#define nl "\n"
#define sp << " " <<

/********* DEBUG *********/

const ll MOD = 1000000007;
const ll inf = 1e18;
const ll mxN = 1000005;

// st s = .
string solve_puzzle(string s, vector<int> c) {
    ll n = sz(s);

    // 0 = X, 1 = _, 2 = ?
    vector<ll> vals(n);
    vector<ll> pref(n), suff(n);

    ll sum = 0;
    for (auto &x : c)
        sum += x;
    sum += sz(c)-1;

    ll ptr = 0, len = 0;
    for (int i = 0; i < n; i++){
        pref[i] = ptr;
        if (s[i] == '.'){
            if (ptr < sz(c) && ++len == c[ptr])
                ptr++, len = -1;
        }
    }

    //outvec(pref);
    ptr = sz(c)-1, len = 0;
    for (int i = n-1; i >= 0; i--){
        suff[i] = sz(c)-1-ptr;
        if (s[i] == '.'){
            if (ptr >= 0 && ++len == c[ptr])
                ptr--, len = -1;
        }
    }

    //outvec(suff);
    string ans = "";
    for (int i = 0; i < n; i++){
        if (pref[i] + suff[i] >= sz(c))
            ans+=(sum == n ? '_' : '?');
        else
            ans +='X';
    }

    return ans;
}

Compilation message (stderr)

paint.h:1:9: warning: #pragma once in main file
    1 | #pragma once
      |         ^~~~
paint_c.h:1:9: warning: #pragma once in main file
    1 | #pragma once
      |         ^~~~

• Subtask 1: 7 / 7
• Subtask 2: 3 / 3
• Subtask 3: 22 / 22
• Subtask 4: 0 / 27
• Subtask 5: 0 / 21
• Subtask 6: 0 / 10
• Subtask 7: 0 / 10