Language: C++

#include <bits/stdc++.h>
#define REP(i,n) for(int i=0;i<n;i++)
#define REPP(i,n) for(int i=1;i<=n;i++)
const double PI = acos(-1);
const double EPS = 1e-15;
long long INF=(long long)1E17;
#define i_7 (long long)(1E9+7)
long mod(long a){
    long long c=a%i_7;
    if(c>=0)return c;
    return c+i_7;
}
using namespace std;
bool prime(int n){
  if(n==1){
    return false;
  }else if(n==2){
    return true;
  }else{
    for(int i=2;i<=sqrt(n);i++){
      if(n%i==0){
        return false;
      }
    }
    return true;
  }
}

long long gcd(long long a, long long b){
  if(a<b){
    swap(a,b);
  }
  if(a%b==0){
    return b;
  }else{
    return gcd(b,a%b);
  }
}
 
long long lcm(long long x, long long y){
  return (x/gcd(x,y))*y;
}
 
class UnionFind {
    public:
    //各頂点の親の番号を格納する。その頂点自身が親だった場合は-(その集合のサイズ)を入れる。
    vector<int> Parent;
    
    //クラスを作るときは、Parentの値を全て-1にする。
    //以下のようにすると全てバラバラの頂点として解釈できる。
    UnionFind(int N) {
        Parent = vector<int>(N, -1);
    }
    
    //Aがどのグループに属しているか調べる
    int root(int A) {
        if (Parent[A] < 0) return A;
        return Parent[A] = root(Parent[A]);
    }
    
    //自分のいるグループの頂点数を調べる
    int size(int A) {
        return -Parent[root(A)];//先祖をrootで取っておきたい。
    }
    
    //AとBをくっ付ける
    bool connect(int A, int B) {
        //AとBを直接つなぐのではなく、root(A)にroot(B)をくっつける
        A = root(A);
        B = root(B);
        if (A == B) {
            //すでにくっついてるからくっ付けない
            return false;
        }
        
        //大きい方(A)に小さいほう(B)をくっ付けたい
        //大小が逆だったらAとBをひっくり返す。
        if (size(A) < size(B)) swap(A, B);
        
        //Aのサイズを更新する
        Parent[A] += Parent[B];
        //Bの親をAに変更する
        Parent[B] = A;
        
        return true;
    }
};

long long dfs(int prev,int now,long long k,vector<int> &G,vector<int> &used){
  int temp = G[now].size();
  long long res;
  if(prev==-1){
    res = mod(k);
    used[now]=true;
    REP(i,temp){
      res = mod(res*dfs(now,G[now][i],k,G,used));
    }
  }else{
    int adj = G[prev].size();
    long long no_use=0;
    REP(i,adj){
      if(G[prev][i]!=now && used[G[prev][i]]){
        no_use++;
      }
    }
    if(k-no_use-1>0){
      res = mod(k-no_use-1);
    }else{
      res = 0;
    }
    used[now] = true;
    
    REP(i,temp){
      if(G[now][i]==prev)continue;
      res = mod(res*dfs(now,G[now][i],k,G,used));
    }
  }
  return res;
}
    
  
int main(){
  int n;
  long long k;
  cin>>n>>k;
  
  vector<int> G[n];
  int a,b;
  REP(i,n-1){
    cin>>a>>b;
    a--;b--;
    G[a].push_back(b);
    G[b].push_back(a);
  }
  
  bool used[n]={};
  
  long long ans;
  ans = dfs(-1,0,k,G,used);
  cout<<ans<<endl;
  return 0;
}



Compiler output:
a.cc: In function 'long long int dfs(int, int, long long int, std::vector<int>&, std::vector<int>&)':
a.cc:90:21: error: request for member 'size' in '(& G)->std::vector<int>::operator[](((std::vector<int>::size_type)now))', which is of non-class type '__gnu_cxx::__alloc_traits<std::allocator<int>, int>::value_type' {aka 'int'}
   90 |   int temp = G[now].size();
      |                     ^~~~
a.cc:96:35: error: invalid types '__gnu_cxx::__alloc_traits<std::allocator<int>, int>::value_type {aka int}[int]' for array subscript
   96 |       res = mod(res*dfs(now,G[now][i],k,G,used));
      |                                   ^
a.cc:99:23: error: request for member 'size' in '(& G)->std::vector<int>::operator[](((std::vector<int>::size_type)prev))', which is of non-class type '__gnu_cxx::__alloc_traits<std::allocator<int>, int>::value_type' {aka 'int'}
   99 |     int adj = G[prev].size();
      |                       ^~~~
a.cc:102:17: error: invalid types '__gnu_cxx::__alloc_traits<std::allocator<int>, int>::value_type {aka int}[int]' for array subscript
  102 |       if(G[prev][i]!=now && used[G[prev][i]]){
      |                 ^
a.cc:102:41: error: invalid types '__gnu_cxx::__alloc_traits<std::allocator<int>, int>::value_type {aka int}[int]' for array subscript
  102 |       if(G[prev][i]!=now && used[G[prev][i]]){
      |                                         ^
a.cc:114:16: error: invalid types '__gnu_cxx::__alloc_traits<std::allocator<int>, int>::value_type {aka int}[int]' for array subscript
  114 |       if(G[now][i]==prev)continue;
      |                ^
a.cc:115:35: error: invalid types '__gnu_cxx::__alloc_traits<std::allocator<int>, int>::value_type {aka int}[int]' for array subscript
  115 |       res = mod(res*dfs(now,G[now][i],k,G,used));
      |                                   ^
a.cc: In function 'int main()':
a.cc:139:20: error: invalid initialization of reference of type 'std::vector<int>&' from expression of type 'std::vector<int> [n]'
  139 |   ans = dfs(-1,0,k,G,used);
      |                    ^
a.cc:89:57: note: in passing argument 4 of 'long long int dfs(int, int, long long int, std::vector<int>&, std::vector<int>&)'
   89 | long long dfs(int prev,int now,long long k,vector<int> &G,vector<int> &used){
      |                                            ~~~~~~~~~~~~~^