oj.uz

Submission #791575

# Submission time Handle Problem Language Result Execution time Memory
791575 2023-07-24T07:37:49 Z ttamx Star Trek (CEOI20_startrek) C++14
7 / 100
 4 ms 5204 KB 
#include<bits/stdc++.h>

using namespace std;

typedef long long ll;
typedef long double ld;

const int N=1e5+5;
const ll mod=1e9+7;

ll binpow(ll a,ll b){
    ll res=1;
    while(b>0){
        if(b&1)res*=a,res%=mod;
        a*=a,a%=mod;
        b>>=1;
    }
    return res;
}

int n;
ll d,ans,sum;
vector<int> adj[N];
ll dp[N][4];
bool dp1[N],dp2[N];

void dfs(int u,int p){
    for(auto v:adj[u]){
        if(v==p)continue;
        dfs(v,u);
        dp1[u]|=!dp1[v];
    }
}

void dfs2(int u,int p,bool w){
    int cnt=!w;
    dp2[u]=dp1[u]|!w;
    for(auto v:adj[u]){
        if(v==p)continue;
        cnt+=!dp1[v];
    }
    for(auto v:adj[u]){
        if(v==p)continue;
        dfs2(v,u,cnt-!dp1[v]>0);
    }
}

ll dfs3(int u,int p){
    int cnt=0;
    for(auto v:adj[u]){
        if(v==p)continue;
        cnt+=!dp1[v];
    }
    for(auto v:adj[u]){
        if(v==p)continue;
        dfs3(v,u);
        dp[u][0]+=dp[v][1];
        if(cnt-!dp1[v]>0)dp[u][0]+=dp[v][0];
        else dp[u][1]+=dp[v][0];
        dp[u][0]%=mod;
        dp[u][1]%=mod;
    }
    dp[u][0]+=n-sum;
    if(dp1[u])dp[u][0]+=sum;
    else dp[u][1]+=sum;
    dp[u][0]%=mod;
    dp[u][1]%=mod;
}

int main(){
    cin.tie(nullptr)->sync_with_stdio(false);
    cin >> n >> d;
    if(n==2)cout << binpow(4,d),exit(0);
    for(int i=1;i<n;i++){
        int u,v;
        cin >> u >> v;
        adj[u].emplace_back(v);
        adj[v].emplace_back(u);
    }
    dfs(1,1);
    dfs2(1,1,true);
    for(int i=1;i<=n;i++)sum+=dp2[i];
    dfs3(1,1);
    cout << dp[1][0];
}

Compilation message

startrek.cpp: In function 'll dfs3(int, int)':
startrek.cpp:68:1: warning: no return statement in function returning non-void [-Wreturn-type]
   68 | }
      | ^

Subtask #1
0 / 0.0

# Verdict Execution time Memory Grader output
1 Runtime error 4 ms 5204 KB Execution killed with signal 11
2 Halted 0 ms 0 KB -

Subtask #2
7.0 / 7.0

# Verdict Execution time Memory Grader output
1 Correct 2 ms 2644 KB Output is correct
2 Correct 1 ms 2644 KB Output is correct
3 Correct 1 ms 2644 KB Output is correct
4 Correct 1 ms 2644 KB Output is correct
5 Correct 1 ms 2644 KB Output is correct

Subtask #3
0 / 8.0

# Verdict Execution time Memory Grader output
1 Runtime error 4 ms 5196 KB Execution killed with signal 11
2 Halted 0 ms 0 KB -

Subtask #4
0 / 15.0

# Verdict Execution time Memory Grader output
1 Runtime error 4 ms 5196 KB Execution killed with signal 11
2 Halted 0 ms 0 KB -

Subtask #5
0 / 15.0

# Verdict Execution time Memory Grader output
1 Runtime error 4 ms 5196 KB Execution killed with signal 11
2 Halted 0 ms 0 KB -

Subtask #6
0 / 20.0

# Verdict Execution time Memory Grader output
1 Runtime error 4 ms 5196 KB Execution killed with signal 11
2 Halted 0 ms 0 KB -

Subtask #7
0 / 20.0

# Verdict Execution time Memory Grader output
1 Runtime error 4 ms 5196 KB Execution killed with signal 11
2 Halted 0 ms 0 KB -

Subtask #8
0 / 15.0

# Verdict Execution time Memory Grader output
1 Runtime error 4 ms 5204 KB Execution killed with signal 11
2 Halted 0 ms 0 KB -