AtCoder|ABC046:B - AtCoDeerくんとボール色塗り

AtCoderの過去問対策です。ABCで解けなかった問題、ためになった問題のコードを備忘録として残します。

問題

B - Painting Balls with AtCoDeer

解説

https://img.atcoder.jp/data/arc/062/editorial.pdf

解答例

#include <bits/stdc++.h>
using namespace std;

int main(){
  int N, K;
  cin >> N >> K;

  int ans = K;

  // K * (K − 1)N−1乗 通り
  for(int i = 1; i < N; i++){
    ans *= K-1;
  }  
  cout << ans << "\n";  
}

メモ

• K * (K − 1)N−1乗 通りを計算する