AIM Tech Round 3 (Div. 2) -- D. Recover the String (構造字符串)

題目
題意:
給你一個只包含01的字符串,告訴你00,01,10,11子字符串的個數,還原這個字符串。不能存在則輸出Impossible.

Examples
input
1 2 3 4
output
Impossible
input
1 2 2 1
output
0110

從00和11的個數,可以算出字符串中0和1的個數,然後即可構造。但當a00,a11中有爲0的情況下,需要特殊考慮(代碼水平問題….各種情況寫的亂了,最後還是沒有a….)。

#include <iostream>
#include <cmath>

using namespace std;

long long a00, a01, a10, a11;

int main() {
    bool flag = true;
    cin >> a00 >> a01 >> a10 >> a11;
    long long o = -1, z = -1;
    for(long long i = 1; i * (i - 1) / 2 <= (a00+1); i++){
        if(i * (i - 1) == a00 * 2){
            z = i;
        }
    }
    for(long long i = 1; i * (i - 1) / 2 <= (a11+1); i++){
        if(i * (i - 1) == a11 * 2){
            o = i;
        }
    }
    if(a00 == 0 && a01 == 0 && a10 == 0 && a11 == 0){
        cout << '1' << endl;
        return 0;
    }
    if(a00 == 0){
        if(a01 || a10) z = 1;
        else z = 0;
    }

    if(a11 == 0){
        if(a10 || a01) o = 1;
        else o = 0;
    }

    if(o == -1 || z == -1)
        flag = false;

    if(a01 > z * o || a10 > z * o)
        flag = false;

    if(a01 + a10 != z * o)  
        flag = false;

    if(flag) {
        long long t1 = 0, t2 = 0;
        while(o + z) {
            if(t1 + o > a01){
                cout << '1';
                a10 -= z;
                o--;
            }
            else{
                cout << '0';
                t1 += o;
                z--;
            }
        }
    }
    else {
        cout << "Impossible";
    }
    cout << endl;
    return 0;
}
發佈了68 篇原創文章 · 獲贊 23 · 訪問量 3萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章