mayoko’s diary

プロコンとかいろいろ。

yukicoder No.297 カードの数式

yukicoder に参加しました。初めて作問して出題したのですが, 想定解が間違っていて申し訳ないです…

解法

一つ巨大な数を作ってそれ以外の数をひと桁にすると最大, 最小の数が作れそうです。
ただ, マイナス演算子がひとつもない場合は, 巨大な数をつくらず各数の桁を分割するのが良いです。大きい数を小さい桁に置いていきます。

int main() {
    cin.tie(0);
    ios::sync_with_stdio(false);
    int N;
    cin >> N;
    string s;
    s.resize(N);
    vector<int> vs;
    int minuscnt = 0, pluscnt = 0;
    for (int i = 0; i < N; i++) {
        cin >> s[i];
        if (s[i] == '+') pluscnt++;
        else if (s[i] == '-') minuscnt++;
        else vs.push_back(s[i]-'0');
    }
    sort(vs.begin(), vs.end());
    reverse(vs.begin(), vs.end());
    ll maxi = 0;
    int cnt = vs.size()-pluscnt-minuscnt;
    for (int i = 0; i < cnt; i++) {
        maxi = maxi*10+vs[i];
    }
    {
        ll ans = maxi;
        for (int i = 0; i < pluscnt; i++) {
            ans += vs[i+cnt];
        }
        for (int i = 0; i < minuscnt; i++) {
            ans -= vs[i+cnt+pluscnt];
        }
        cout << ans << " ";
    }
    {
        if (minuscnt) {
            ll ans = -maxi;
            for (int i = 0; i < minuscnt-1; i++) {
                ans -= vs[i+cnt];
            }
            for (int i = 0; i < pluscnt+1; i++) {
                ans += vs[i+cnt+minuscnt-1];
            }
            cout << ans << endl;
        } else {
            ll ans = 0;
            int p = pluscnt+minuscnt+1;
            int cur = 0, tmp = 0;
            while (cur < N) {
                for (int i = 0; i < p; i++) {
                    if (cur >= N) break;
                    ll hoge = vs[cur++];
                    for (int j = 0; j < tmp; j++) hoge *= 10;
                    ans += hoge;
                }
                tmp++;
            }
            cout << ans << endl;
        }
    }
    return 0;
}