采用公式推导法

原理：乘法原理

结论：有$n$个球，就有$k \times (k+1)^{n-1}$种染法。

AC Code

#include<bits/stdc++.h>
using namespace std;
int main()
{
     int n,k;
     cin >> n >> k;
     int ans; 
     ans = k * pow(k - 1, n - 1);
     cout << ans << endl;
     return 0;
}