422021-01-08 20:34:25mraronStar Trekcpp11Elfogadva 100/100118ms45360 KiB
#include <iostream>
#include <vector>

using namespace std;

const long long mod = 1000000007;
long long N, D, a, b, L, W, LC, WC, sL, sW, sLC, sWC;
vector<int> el[100000];
bool lose[100000];
bool volt[100000];
long long c[100000];
long long oW[100000];
long long oL[100000];
long long h[100000];

void be(int x){
    volt[x] = true; lose[x] = true;
    for(int i:el[x]){
        if(!volt[i]){
            be(i);
            if(lose[i]) h[x]++;
        }
    }
    if(h[x]!=0) lose[x] = false;
}
void ertek(int x){
    if(lose[x]){
        c[x] = oW[x] + 1;
    } else {
        if(h[x]==1){
            c[x] = oL[x];
        } else {
            c[x] = 0;
        }
    }
}
void ker(int x){
    volt[x] = true;
    for(int i:el[x]){
        if(!volt[i]){
            ker(i);
            if(lose[i]){
                oL[x] += c[i];
            } else {
                oW[x] += c[i];
            }
        }
    }
    ertek(x);
}
void reroot(int x, int y){
    if(lose[y]){
        h[x]--; oL[x] -= c[y];
    } else {
        oW[x] -= c[y];
    }
    if(h[x]==0) lose[x] = true;
    ertek(x);
    if(lose[x]){
        h[y]++; oL[y] += c[x];
    } else {
        oW[y] += c[x];
    }
    if(h[y]!=0) lose[y] = false;
    ertek(y);
}
void lep(int x){
    volt[x] = true;
    if(lose[x]){
        L++; LC = (LC+c[x])%mod;
    } else {
        W++; WC = (WC+c[x])%mod;
    }
    for(int i:el[x]){
        if(!volt[i]){
            reroot(x,i);
            lep(i);
            reroot(i,x);
        }
    }
}

void step(long long pl, long long plc, long long pw, long long pwc){
    sL = ((((pl * N) % mod) * W) % mod + (pwc * L) % mod + ((((pl * N)%mod - plc + mod)%mod) * L)%mod) % mod;
    sLC = (plc * WC + pwc * LC) % mod;
    sW = ((((pw * N) % mod) * W) % mod + (plc * L) % mod + ((((pw * N)%mod - pwc + mod)%mod) * L)%mod) % mod;
    sWC = (pwc * WC + plc * LC) % mod;
}
void step2(long long pl, long long plc, long long pw, long long pwc){
    L = ((((pl * N) % mod) * pw) % mod + (pwc * pl) % mod + ((((pl * N)%mod - plc + mod)%mod) * pl)%mod) % mod;
    LC = (plc * pwc + pwc * plc) % mod;
    W = ((((pw * N) % mod) * pw) % mod + (plc * pl) % mod + ((((pw * N)%mod - pwc + mod)%mod) * pl)%mod) % mod;
    WC = (pwc * pwc + plc * plc) % mod;
}

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

    cin >> N >> D;
    for(int i=0; i<N-1; i++){
        cin >> a >> b;
        el[a-1].push_back(b-1);
        el[b-1].push_back(a-1);
    }

    be(0);
    for(int i=0; i<N; i++) volt[i] = false;
    ker(0);
    for(int i=0; i<N; i++) volt[i] = false;
    lep(0);

    if(lose[0]){
        sL = 1; sLC = c[0];
    } else {
        sW = 1; sWC = c[0];
    }

    //cout << L << ' ' << LC << ' ' << W << ' ' << WC << endl;

    while(D>0){
        if(D%2==1) step(sL, sLC, sW, sWC);
        step2(L,LC,W,WC); D /= 2;
    }

    cout << sW << '\n';

    return 0;
}
RészfeladatÖsszpontTesztVerdiktIdőMemória
subtask10/0
1Elfogadva4ms6492 KiB
2Elfogadva4ms6712 KiB
subtask27/7
3Elfogadva4ms6588 KiB
4Elfogadva3ms6592 KiB
5Elfogadva3ms6596 KiB
6Elfogadva3ms6600 KiB
7Elfogadva3ms6600 KiB
subtask38/8
8Elfogadva3ms6620 KiB
9Elfogadva3ms6628 KiB
10Elfogadva3ms6628 KiB
11Elfogadva3ms6632 KiB
12Elfogadva3ms6636 KiB
subtask415/15
13Elfogadva4ms6864 KiB
14Elfogadva4ms6908 KiB
15Elfogadva4ms6864 KiB
16Elfogadva4ms6864 KiB
17Elfogadva4ms6876 KiB
subtask515/15
18Elfogadva103ms26664 KiB
19Elfogadva108ms34652 KiB
20Elfogadva78ms23412 KiB
21Elfogadva97ms24568 KiB
22Elfogadva97ms25824 KiB
subtask620/20
23Elfogadva4ms12940 KiB
24Elfogadva4ms13112 KiB
25Elfogadva4ms12976 KiB
26Elfogadva4ms12996 KiB
27Elfogadva3ms13004 KiB
subtask720/20
28Elfogadva115ms36332 KiB
29Elfogadva103ms38484 KiB
30Elfogadva59ms27880 KiB
31Elfogadva104ms30484 KiB
32Elfogadva104ms31724 KiB
subtask815/15
33Elfogadva112ms45360 KiB
34Elfogadva118ms44084 KiB
35Elfogadva61ms32588 KiB
36Elfogadva107ms42024 KiB
37Elfogadva100ms37672 KiB
38Elfogadva89ms38892 KiB
39Elfogadva97ms40080 KiB
40Elfogadva93ms41344 KiB