Kierrokset

input/code.cpp: In function 'void haku(int, std::vector<int>)':
input/code.cpp:33:34: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     if(solmu == y && path.size() == lenMin){
                      ~~~~~~~~~~~~^~~~~~~~~
input/code.cpp:34:23: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
         for(int i=0; i<path.size(); i++)path1.push_back(path[i]);
                      ~^~~~~~~~~~~~
input/code.cpp:42:23: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
         for(int i=0; i<path.size(); i++)newPath.push_back(path[i]);
                      ~^~~~~~~~~~~~

#include <bits/stdc++.h>

using namespace std;
int n, x, y,a,b;

long long ans=0;

long long MOD = pow(10,9)+7;

int lenMin = INT_MAX;

vector<int> path1;

vector<int> v[1010101];

int z[1010101];

void hakuLen(int solmu, int len){
    if(solmu == y){
        lenMin = min(lenMin, len);
    }
    if(z[solmu]) return;

    z[solmu]=1;

    for(auto s : v[solmu]){
        hakuLen(s,len+1);
    }
}

void haku(int solmu, vector<int> path){

    if(solmu == y && path.size() == lenMin){
        for(int i=0; i<path.size(); i++)path1.push_back(path[i]);
    }
    if(z[solmu]) return;

    z[solmu]=1;

    for(auto s : v[solmu]){
        vector<int> newPath;
        for(int i=0; i<path.size(); i++)newPath.push_back(path[i]);
        newPath.push_back(s);
        haku(s,newPath);
    }
}

int factorial(int n){
    if(n == 0) return 1;
    return n*factorial(n-1)%MOD;
}

int calc(int n, int r){

    return factorial(n)/(factorial(r)*factorial(n-r));
}

int main(){
    cin>>n>>x>>y;

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

    hakuLen(x, 1);

    //haku(x,);
    lenMin--;



    int p = n-1-lenMin;

    for(int i=0; i<=p; i++){
        ans += calc(p, i)%MOD;
    }


    cout<<ans<<"\n";

    return 0;
}

Test 1

Group: 1, 2, 4

Verdict: WRONG ANSWER

Test 2

Group: 1, 4

Verdict: WRONG ANSWER

Test 3

Group: 1, 4

Verdict: WRONG ANSWER

Test 4

Group: 1, 4

Verdict: WRONG ANSWER

Test 5

Group: 2, 4

Verdict: WRONG ANSWER

Test 6

Group: 4

Verdict: WRONG ANSWER

Test 7

Group: 4

Verdict: WRONG ANSWER

Test 8

Group: 4

Verdict: WRONG ANSWER

Test 9

Group: 2, 4

Verdict: WRONG ANSWER

Test 10

Group: 4

Verdict: WRONG ANSWER

Test 11

Group: 4

Verdict: WRONG ANSWER

Test 12

Group: 4

Verdict: WRONG ANSWER

Test 13

Group: 2, 4

Verdict: WRONG ANSWER

Test 14

Group: 4

Verdict: WRONG ANSWER

Test 15

Group: 4

Verdict: WRONG ANSWER

Test 16

Group: 4

Verdict: WRONG ANSWER

Test 17

Group: 2, 4

Verdict: WRONG ANSWER

Test 18

Group: 4

Verdict: TIME LIMIT EXCEEDED

Test 19

Group: 4

Verdict: TIME LIMIT EXCEEDED

Test 20

Group: 4

Verdict: TIME LIMIT EXCEEDED

Test 21

Group: 2, 4

Verdict: TIME LIMIT EXCEEDED

Test 22

Group: 4

Verdict: TIME LIMIT EXCEEDED

Test 23

Group: 4

Verdict: TIME LIMIT EXCEEDED

Test 24

Group: 4

Verdict: TIME LIMIT EXCEEDED

Test 25

Group: 2, 4

Verdict: TIME LIMIT EXCEEDED

Test 26

Group: 4

Verdict: TIME LIMIT EXCEEDED

Test 27

Group: 4

Verdict: TIME LIMIT EXCEEDED

Test 28

Group: 4

Verdict: TIME LIMIT EXCEEDED

Test 29

Group: 1, 2, 3, 4

Verdict: ACCEPTED

Test 30

Group: 1, 2, 3, 4

Verdict: ACCEPTED

Test 31

Group: 1, 2, 3, 4

Verdict: ACCEPTED

Test 32

Group: 1, 2, 4

Verdict: ACCEPTED

Test 33

Group: 1, 2, 3, 4

Verdict: ACCEPTED

Test 34

Group: 1, 2, 4

Verdict: ACCEPTED

Test 35

Group: 1, 2, 4

Verdict: ACCEPTED

Test 36

Group: 1, 3, 4

Verdict: WRONG ANSWER

Test 37

Group: 1, 4

Verdict: WRONG ANSWER

Test 38

Group: 1, 4

Verdict: WRONG ANSWER

Test 39

Group: 1, 4

Verdict: ACCEPTED

Test 40

Group: 1, 4

Verdict: WRONG ANSWER

Test 41

Group: 1, 4

Verdict: WRONG ANSWER

Test 42

Group: 1, 4

Verdict: WRONG ANSWER

Test 43

Group: 1, 4

Verdict: WRONG ANSWER

Test 44

Group: 1, 4

Verdict: WRONG ANSWER

Test 45

Group: 1, 4

Verdict: WRONG ANSWER

Test 46

Group: 1, 2, 4

Verdict: ACCEPTED

Test 47

Group: 1, 2, 3, 4

Verdict: WRONG ANSWER

Test 48

Group: 1, 2, 3, 4

Verdict: WRONG ANSWER

Test 49

Group: 1, 2, 4

Verdict: ACCEPTED

Test 50

Group: 1, 4

Verdict: ACCEPTED

Test 51

Group: 2, 4

Verdict: ACCEPTED

Test 52

Group: 2, 3, 4

Verdict: TIME LIMIT EXCEEDED

Test 53

Group: 2, 3, 4

Verdict: TIME LIMIT EXCEEDED

Test 54

Group: 2, 3, 4

Verdict: TIME LIMIT EXCEEDED

Test 55

Group: 2, 4

Verdict: TIME LIMIT EXCEEDED

Test 56

Group: 2, 4

Verdict: TIME LIMIT EXCEEDED

Test 57

Group: 2, 4

Verdict: TIME LIMIT EXCEEDED

Test 58

Group: 2, 4

Verdict: TIME LIMIT EXCEEDED

Test 59

Group: 2, 4

Verdict: TIME LIMIT EXCEEDED

Test 60

Group: 2, 4

Verdict: TIME LIMIT EXCEEDED

Test 61

Group: 2, 4

Verdict: TIME LIMIT EXCEEDED

Test 62

Group: 2, 4

Verdict: TIME LIMIT EXCEEDED

Test 63

Group: 4

Verdict: TIME LIMIT EXCEEDED

Test 64

Group: 4

Verdict: TIME LIMIT EXCEEDED

Test 65

Group: 4

Verdict: TIME LIMIT EXCEEDED

Test 66

Group: 4

Verdict: TIME LIMIT EXCEEDED

Test 67

Group: 4

Verdict: TIME LIMIT EXCEEDED

Test 68

Group: 4

Verdict: TIME LIMIT EXCEEDED

Test 69

Group: 4

Verdict: TIME LIMIT EXCEEDED

Test 70

Group: 4

Verdict: TIME LIMIT EXCEEDED

Test 71

Group: 4

Verdict: TIME LIMIT EXCEEDED

Test 72

Group: 4

Verdict: TIME LIMIT EXCEEDED

Test 73

Group: 3, 4

Verdict: TIME LIMIT EXCEEDED

Test 74

Group: 3, 4

Verdict: TIME LIMIT EXCEEDED

Test 75

Group: 3, 4

Verdict: TIME LIMIT EXCEEDED

Test 76

Group: 3, 4

Verdict: TIME LIMIT EXCEEDED

Test 77

Group: 3, 4

Verdict: TIME LIMIT EXCEEDED

Test 78

Group: 3, 4

Verdict: TIME LIMIT EXCEEDED

Test 79

Group: 3, 4

Verdict: TIME LIMIT EXCEEDED

Test 80

Group: 3, 4

Verdict: TIME LIMIT EXCEEDED

Test 81

Group: 3, 4

Verdict: TIME LIMIT EXCEEDED

Test 82

Group: 3, 4

Verdict: TIME LIMIT EXCEEDED

Test 83

Group: 2, 4

Verdict: TIME LIMIT EXCEEDED

Test 84

Group: 2, 4

Verdict: TIME LIMIT EXCEEDED

Test 85

Group: 1, 2, 4

Verdict: WRONG ANSWER

Test 86

Group: 1, 2, 4

Verdict: ACCEPTED