【競プロ精進日記】その他編＜５＞Beginaid

本記事は管理人が競技プログラミングを始めたため，その精進日記としてログを取ったものです。アウトプットして積極的にモチベーションを上げていく作戦です。記事目次は以下をご覧ください。

内容は管理人の推測や恣意的な感想を大いに含みます。もし間違い等ありましたらご指摘ください。あくまでも参考程度にお願い致します。

【競プロ精進日記】目次まとめ ABC Atcoder Beginner Contestの過去問を解いています。最低限のコメントと一言が添えられています。 ...

その他の問題

過去問精選10問でその他の問題に分類されているものに取り組んでいきます。今回は簡単な数学に関する問題（ABC 046 B）です。使用言語はc++です。

c++

#include <bits/stdc++.h>
#define _GLIBCXX_DEBUG
#define rep(i, n) for (int i = 0; i < (int)(n); i++)
#define repi(i, a, b) for(int i = (int)(a); i < (int)(b); i++)
using namespace std;

int main() {
  int N=0, K=0;
  cin >> N >> K;
  long long ans = K;
  rep(n, N-1){
    ans *= (K-1);
  }
  cout << ans << endl;
}

中学数学ですね。最初にK種類の色から1つを選び，そのあとは1つ手前で選んだ色以外のK-1種類の色から1つを選べばOKです。