제출 #428379

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
428379		Ozy	Star Trek (CEOI20_startrek)	C++17	0 / 100	15 ms	2676 KiB

startrek

#include <iostream>
#include <bits/stdc++.h>
using namespace std;
#define lli long long int
#define rep(i,a,b) for(int i = (a); i <= (b); i++)
#define repa(i,a,b) for(int i = (a); i >= (b); i--)
#define debugsl(a) cout << #a << " = " << a << ", "
#define debug(a) cout << #a << " = " << a << endl;

#define MAX 100000
#define mod 1000000007

bool si;
lli n,d,a,b,total,g,p;
vector<lli> hijos[MAX+2];
lli estado[MAX+2],original[MAX+2];

lli DFS (lli pos, lli padre) {

    lli res = 1;
    for (auto h : hijos[pos]) {
        if (h == padre) continue;
        res *= DFS(h,pos);
    }

    if (res == 0) {
        if (si) original[pos] = 1;
        return 1;
    }
    else {
        if (si) original[pos] = 0;
        return 0;
    }
}

void resuelve(lli pos, lli padre, lli prof) {

    if (prof%2 == 0) total += p;
    else if (original[pos] == 0) total += g;

    if (total > mod) total %= mod;

    lli ap = 0;
    lli ag = 0;
    for(auto h : hijos[pos]) {
        if (h == padre) continue;
        if (original[h] == 0) ap++;
        else ag++;
    }

    if (prof%2 && original[pos] == 1 && ap > 1) return;

    for(auto h : hijos[pos]) {
        if (h == padre) continue;

        if (prof%2 && original[pos] == 1 && original[h] == 0) resuelve(h,pos,prof+1);
        else if (prof%2 && original[pos] == 0) resuelve(h,pos,prof+1);
        else if (!(prof%2)) resuelve(h,pos,prof+1);
    }
}

int main()
{
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);

    cin >> n >> d;
    rep(i,1,n-1) {
        cin >> a >> b;
        hijos[a].push_back(b);
        hijos[b].push_back(a);
    }

    si = true;
    rep(i,1,n) {
        estado[i] = DFS(i,0);

        si = false;
        if (estado[i] == 0) p++;
        else g++;
    }

    total = 0;
    resuelve(1,0,0);

    cout << total;

    return 0;
}

• Subtask 1: 0 / 0
• Subtask 2: 0 / 7
• Subtask 3: 0 / 8
• Subtask 4: 0 / 15
• Subtask 5: 0 / 15
• Subtask 6: 0 / 20
• Subtask 7: 0 / 20
• Subtask 8: 0 / 15