제출 #991183

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
991183		aryan12	앨리스, 밥, 서킷 (APIO23_abc)	C++17	62 / 100	56 ms	11592 KiB

abc

#include "abc.h"
#include <bits/stdc++.h>
using namespace std;
 
// you may find the definitions useful
const int OP_ZERO    = 0;  // f(OP_ZERO,    x0, x1) = 0
const int OP_NOR     = 1;  // f(OP_NOR,     x0, x1) = !(x0 || x1)
const int OP_GREATER = 2;  // f(OP_GREATER, x0, x1) = (x0 > x1)
const int OP_NOT_X1  = 3;  // f(OP_NOT_X1,  x0, x1) = !x1
const int OP_LESS    = 4;  // f(OP_LESS,    x0, x1) = (x0 < x1)
const int OP_NOT_X0  = 5;  // f(OP_NOT_X0,  x0, x1) = !x0
const int OP_XOR     = 6;  // f(OP_XOR,     x0, x1) = (x0 ^ x1)
const int OP_NAND    = 7;  // f(OP_NAND,    x0, x1) = !(x0 && x1)
const int OP_AND     = 8;  // f(OP_AND,     x0, x1) = (x0 && x1)
const int OP_EQUAL   = 9;  // f(OP_EQUAL,   x0, x1) = (x0 == x1)
const int OP_X0      = 10; // f(OP_X0,      x0, x1) = x0
const int OP_GEQ     = 11; // f(OP_GEQ,     x0, x1) = (x0 >= x1)
const int OP_X1      = 12; // f(OP_X1,      x0, x1) = x1
const int OP_LEQ     = 13; // f(OP_LEQ,     x0, x1) = (x0 <= x1)
const int OP_OR      = 14; // f(OP_OR,      x0, x1) = (x0 || x1)
const int OP_ONE     = 15; // f(OP_ONE,     x0, x1) = 1
 
 
// Alice
int // returns la
alice(
    /*  in */ const int n,
    /*  in */ const char names[][5],
    /*  in */ const unsigned short numbers[],
    /* out */ bool outputs_alice[]
) {
    // outputs_alice[0] = 1;
    // outputs_alice[1] = 0;
    // return 2;
    vector<pair<string, int> > names_vector;
    for(int i = 0; i < n; i++) {
        string s = "";
        for(int j = 0; j < 5; j++) {
            if(names[i][j] == '\0') {
                break;
            }
            s += names[i][j];
        }
        names_vector.push_back({s, i});
    }
    sort(names_vector.begin(), names_vector.end());
    int cnt = 0;
    for(int i = 0; i < 30; i++) {
        int xx = names_vector[i].second;
        for(int j = 0; j < 16; j++) {
            if(i >= n) {
                outputs_alice[cnt++] = 0;
            }
            else {
                outputs_alice[cnt++] = ((1 << j) & numbers[xx]) >> j;
            }
        }
    }
    int change[30][30];
    for(int i = 0; i < 30; i++) {
        for(int j = 0; j < 30; j++) {
            if(i >= n || j >= n) {
                change[i][j] = 0;
            }
            else {
                change[i][j] = (names_vector[j].second == i) ? 1 : 0;
            }
            outputs_alice[cnt++] = change[i][j];
        }
    }
    for(int i = 0; i < n; i++) {
        outputs_alice[cnt++] = 0;
    }
    return cnt;
}
 
 
// Bob
int // returns lb
bob(
    /*  in */ const int m,
    /*  in */ const char senders[][5],
    /*  in */ const char recipients[][5],
    /* out */ bool outputs_bob[]
) {
    // outputs_bob[0] = 1;
    // outputs_bob[1] = 1;
    // outputs_bob[2] = 0;
    // return 3;
    map<string, int> names;
    for(int i = 0; i < m; i++) {
        string s = "";
        for(int j = 0; j < 5; j++) {
            if(senders[i][j] == '\0') {
                break;
            }
            s += senders[i][j];
        }
        names[s] = 1;
        s = "";
        for(int j = 0; j < 5; j++) {
            if(recipients[i][j] == '\0') {
                break;
            }
            s += recipients[i][j];
        }
        names[s] = 1;
    }
    int n = names.size();
    int cnt = 0;
    for(auto it: names) {
        names[it.first] = cnt++;
    }
    int a[30][30];
    for(int i = 0; i < 30; i++) {
        for(int j = 0; j < 30; j++) {
            a[i][j] = 0;
        }
    }
    for(int i = 0; i < m; i++) {
        string s = "";
        for(int j = 0; j < 5; j++) {
            if(senders[i][j] == '\0') {
                break;
            }
            s += senders[i][j];
        }
        int num1 = names[s];
        s = "";
        for(int j = 0; j < 5; j++) {
            if(recipients[i][j] == '\0') {
                break;
            }
            s += recipients[i][j];
        }
        int num2 = names[s];
        a[num1][num2] = 1;
    }
    for(int i = 0; i < 30; i++) {
        for(int j = 0; j < 30; j++) {
            outputs_bob[30 * i + j] = a[i][j];
        }
    }
    return 30 * 30;
}
 
 
// Circuit
int // returns l
circuit(
    /*  in */ const int la,
    /*  in */ const int lb,
    /* out */ int operations[],
    /* out */ int operands[][2],
    /* out */ int outputs_circuit[][16]
) {
    // operations[5] = 8;
    // operations[6] = 14;
    // operands[5][0] = 0; operands[5][1] = 4;
    // operands[6][0] = 2; operands[6][1] = 5;
    // int final_results[] = {20000, 0, 24464};
    // for(int i = 0; i < 3; ++i)
    //     for(int j = 0; j < 16; ++j)
    //         outputs_circuit[i][j] = 5 + (final_results[i] >> j & 1);
    // return 7;
 
    int cnt = la + lb;
    int n = la - 30 * 16 - 30 * 30;
 
    auto full_adder = [&](vector<int> prev_ans, vector<int> add_what, int la, int lb, int carry, int return_what) -> vector<int> {
        vector<int> answers_this_time(16);
        for(int j = 0; j < 16; j++) {
            operations[cnt] = OP_XOR;
            operands[cnt][0] = prev_ans[j];
            operands[cnt][1] = add_what[j];
            cnt += 1;
 
            operations[cnt] = OP_XOR;
            operands[cnt][0] = carry;
            operands[cnt][1] = cnt - 1;
            answers_this_time[j] = cnt;
            cnt += 1;
 
            operations[cnt] = OP_OR;
            operands[cnt][0] = prev_ans[j];
            operands[cnt][1] = add_what[j];
            cnt += 1;
 
            operations[cnt] = OP_AND;
            operands[cnt][0] = carry;
            operands[cnt][1] = cnt - 1;
            cnt += 1;
 
            operations[cnt] = OP_AND;
            operands[cnt][0] = prev_ans[j];
            operands[cnt][1] = add_what[j];
            cnt += 1;
 
            operations[cnt] = OP_OR;
            operands[cnt][0] = cnt - 2;
            operands[cnt][1] = cnt - 1;
            carry = cnt;
            cnt += 1;
        }
 
        for(int j = 0; j < 16; j++) {
            // if cur bit is 0, keep prev_ans
            // cout << "cnt = " << cnt << "\n";
            operations[cnt] = OP_GREATER;
            operands[cnt][0] = prev_ans[j];
            operands[cnt][1] = return_what;
            cnt += 1;
 
            // if cur bit is 1, keep new ans
            operations[cnt] = OP_AND;
            operands[cnt][0] = answers_this_time[j];
            operands[cnt][1] = return_what;
            cnt += 1;
 
            operations[cnt] = OP_OR;
            operands[cnt][0] = cnt - 2;
            operands[cnt][1] = cnt - 1;
            prev_ans[j] = cnt;
            cnt += 1;
        }
        return prev_ans;
    };

    auto swap_arrays = [&](vector<int> prev_array, vector<int> cur_array, int to_swap) -> vector<int> {
        vector<int> new_ans(16);
        for(int i = 0; i < 16; i++) {
            operations[cnt] = OP_GREATER;
            operands[cnt][0] = prev_array[i];
            operands[cnt][1] = to_swap;
            cnt += 1;

            operations[cnt] = OP_AND;
            operands[cnt][0] = cur_array[i];
            operands[cnt][1] = to_swap;
            cnt += 1;

            operations[cnt] = OP_OR;
            operands[cnt][0] = cnt - 2;
            operands[cnt][1] = cnt - 1;
            new_ans[i] = cnt;
            cnt += 1;
        }
        return new_ans;
    };
 
    vector<vector<int> > prev_ans(30, vector<int> (16, 0));
    for(int i = 0; i < 30; i++) {
        for(int j = 0; j < 16; j++) {
            operations[cnt] = OP_ZERO;
            operands[cnt][0] = 0;
            operands[cnt][1] = 0;
            prev_ans[i][j] = cnt;
            cnt += 1;
        }
    }
    vector<vector<int> > numbers(30, vector<int> (16, 0));
    for(int i = 0; i < 30; i++) {
        for(int j = 0; j < 16; j++) {
            if(i >= n) {
                operations[cnt] = OP_ZERO;
                operands[cnt][0] = 0;
                operands[cnt][1] = 0;
                numbers[i][j] = cnt;
                cnt += 1;
            }
            else {
                numbers[i][j] = 16 * i + j;
            }
        }
    }
    int carry = cnt;
    operations[cnt] = OP_ZERO;
    operands[cnt][0] = 0;
    operands[cnt][1] = 0;
    cnt += 1;
 
    for(int from = 0; from < 30; from++) {
        for(int to = 0; to < 30; to++) {
            int bob_output = la + from * 30 + to;
            prev_ans[to] = full_adder(prev_ans[to], numbers[from], la, lb, carry, bob_output);
            carry = cnt;
            operations[cnt] = OP_ZERO;
            operands[cnt][0] = 0;
            operands[cnt][1] = 0;
            cnt += 1;
        }
    }

    vector<vector<int> > new_ans(30, vector<int> (16));
    for(int i = 0; i < 30; i++) {
        int current_index = 16 * 30 + 30 * i;
        new_ans[i] = prev_ans[i];
        for(int j = 0; j < 30; j++) {
            if(i >= n || j >= n) {
                continue;
            }
            new_ans[i] = swap_arrays(new_ans[i], prev_ans[j], current_index);
            current_index += 1;
        }
    }

    for(int i = 0; i < n; i++) {
        for(int j = 0; j < 16; j++) {
            outputs_circuit[i][j] = new_ans[i][j];
        }
    }
    return cnt;
}

컴파일 시 표준 에러 (stderr) 메시지

abc.cpp: In function 'int bob(int, const char (*)[5], const char (*)[5], bool*)':
abc.cpp:109:9: warning: unused variable 'n' [-Wunused-variable]
  109 |     int n = names.size();
      |         ^

• Subtask 1: 4 / 4
• Subtask 2: 4 / 4
• Subtask 3: 0 / 4
• Subtask 4: 24 / 24
• Subtask 5: 24 / 24
• Subtask 6: 6 / 6
• Subtask 7: 0 / 18
• Subtask 8: 0 / 10
• Subtask 9: 0 / 6