#672760 (C) No.1477 Lamps on Graph

• 問題: No.1477 Lamps on Graph
• ユーザー: sten_san
• 提出日時: 2021-07-03 03:56:51
• 言語: C (gcc 12.3.0)
• 結果: AC
• 実行時間: 84 ms / 2,000 ms
• コード長: 3,434 bytes
• コンパイル時間: 1,035 ms
• コンパイル使用メモリ: 32,128 KB
• 実行使用メモリ: 10,880 KB
• 最終ジャッジ日時: 2024-06-29 22:00:01
• 合計ジャッジ時間: 4,688 ms
• ジャッジサーバーID （参考情報）: judge1 / judge5

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	1 ms; 6,816 KB
testcase_01	AC	1 ms; 6,940 KB
testcase_02	AC	1 ms; 6,944 KB
testcase_03	AC	1 ms; 6,940 KB
testcase_04	AC	1 ms; 6,940 KB
testcase_05	AC	1 ms; 6,940 KB
testcase_06	AC	1 ms; 6,940 KB
testcase_07	AC	1 ms; 6,940 KB
testcase_08	AC	0 ms; 6,944 KB
testcase_09	AC	1 ms; 6,940 KB
testcase_10	AC	0 ms; 6,944 KB
testcase_11	AC	1 ms; 6,940 KB
testcase_12	AC	43 ms; 6,940 KB
testcase_13	AC	40 ms; 6,940 KB
testcase_14	AC	50 ms; 6,944 KB
testcase_15	AC	23 ms; 6,940 KB
testcase_16	AC	18 ms; 6,944 KB
testcase_17	AC	18 ms; 6,940 KB
testcase_18	AC	51 ms; 9,484 KB
testcase_19	AC	38 ms; 6,940 KB
testcase_20	AC	14 ms; 6,944 KB
testcase_21	AC	41 ms; 8,064 KB
testcase_22	AC	9 ms; 6,944 KB
testcase_23	AC	29 ms; 6,940 KB
testcase_24	AC	60 ms; 7,552 KB
testcase_25	AC	22 ms; 6,944 KB
testcase_26	AC	57 ms; 8,376 KB
testcase_27	AC	27 ms; 6,940 KB
testcase_28	AC	34 ms; 6,940 KB
testcase_29	AC	31 ms; 6,940 KB
testcase_30	AC	26 ms; 6,944 KB
testcase_31	AC	19 ms; 6,944 KB
testcase_32	AC	72 ms; 10,880 KB
testcase_33	AC	71 ms; 10,496 KB
testcase_34	AC	60 ms; 10,880 KB
testcase_35	AC	83 ms; 10,256 KB
testcase_36	AC	84 ms; 10,240 KB
testcase_37	AC	75 ms; 10,368 KB
testcase_38	AC	77 ms; 10,272 KB
testcase_39	AC	80 ms; 10,240 KB

ソースコード

#include <stdio.h>
#include <stdlib.h>
#include <stddef.h>
#include <stdbool.h>

typedef struct vector {
    void *head;
    size_t used, capa;
} *vector;

#define vector_make(...) vector_make_overload(__VA_ARGS__, vector_make_2, vector_make_1)(__VA_ARGS__)
#define vector_make_1(bsize) vector_make_(bsize)
#define vector_make_2(type, size) vector_make_(sizeof(type) * (size))
#define vector_make_overload(arg1, arg2, name, ...) name

#define vector_resize(type, vec, size) vector_resize_(vec, sizeof(type) * (size))

#define vector_index(type, vec, index) ((type *)(vec)->head)[index]
#define vector_size(type, vec) ((vec)->used / sizeof(type))
#define vector_push_back(type, vec, value) \
do { \
    vector_resize(type, vec, vector_size(type, vec) + 1); \
    vector_index(type, vec, vector_size(type, vec) - 1) = value; \
} while (0)

vector vector_make_(size_t bsize) {
    vector v = malloc(sizeof(struct vector));

    if (v == NULL) {
        return NULL;
    }

    v->head = malloc(bsize);

    if (v->head == NULL) {
        free(v);
        return NULL;
    }

    v->used = bsize;
    v->capa = bsize;

    return v;
}

bool vector_resize_(vector v, size_t bsize) {
    if (bsize <= v->capa) {
        v->used = bsize;
        return true;
    }

    size_t new_capa = (bsize + v->capa) / (v->capa + 1) * (v->capa + 1);

    void *new = realloc(v->head, new_capa);
    if (new == NULL) {
        return false;
    }

    v->head = new;
    v->used = bsize;
    v->capa = new_capa;

    return true;
}

vector a;
int compare(const void *x, const void *y) {
    return vector_index(int, a, *(int *)x) - vector_index(int, a, *(int *)y);
}

int main() {
    int n, m;
    scanf("%d %d", &n, &m);

    a = vector_make(int, n);
    for (int i = 0; i < n; ++i) {
        scanf("%d", &vector_index(int, a, i));
    }

    vector g = vector_make(vector, n);
    for (int i = 0; i < n; ++i) {
        vector_index(vector, g, i) = vector_make(int, 0);
    }

    for (int i = 0; i < m; ++i) {
        int u, v;
        scanf("%d %d", &u, &v);
        --u; --v;

        vector_push_back(int, vector_index(vector, g, u), v);
        vector_push_back(int, vector_index(vector, g, v), u);
    }

    vector lamp = vector_make(bool, n);
    for (int i = 0; i < n; ++i) {
        vector_index(bool, lamp, i) = false;
    }

    int k;
    scanf("%d", &k);

    for (int i = 0; i < k; ++i) {
        int b;
        scanf("%d", &b);
        --b;
        vector_index(bool, lamp, b) = true;
    }

    vector x = vector_make(int, 0);

    vector index = vector_make(int, n);
    for (int i = 0; i < n; ++i) {
        vector_index(int, index, i) = i;
    }

    qsort(&vector_index(int, index, 0), n, sizeof(int), &compare);
    for (int i = 0; i < n; ++i) {
        int idx = vector_index(int, index, i);

        if (!vector_index(bool, lamp, idx)) {
            continue;
        }

        vector_push_back(int, x, idx);

        vector list = vector_index(vector, g, idx);

        vector_index(bool, lamp, idx) = false;
        for (int j = 0; j < vector_size(int, list); ++j) {
            int v = vector_index(int, list, j);

            if (vector_index(int, a, idx) < vector_index(int, a, v)) {
                vector_index(bool, lamp, v) ^= 1;
            }
        }
    }

    printf("%zu\n", vector_size(int, x));
    for (int i = 0; i < vector_size(int, x); ++i) {
        printf("%d\n", vector_index(int, x, i) + 1);
    }
}