TCO 2015 Round 1A Med

gistfile1.cpp

#include <bits/stdc++.h>

#define loop(n, i) for(int i=0;i<n;i++)
#define all(v) v.begin(),v.end()

using namespace std;
using ll = long long;
const ll MOD = 1000000007;

class Autogame
{
public:
    int wayscnt(vector <int> a, int K) {
        int N = a.size();
        K = min(N, K);
        vector<int> cnt(N);
        loop (N, i) {
            int p = i+1;
            loop (K, j) p = a[p-1];
            cnt[p-1]++;
        }
        ll ans = 1;
        for (int v : cnt) {
            ans *= v + 1;
            ans %= MOD;
        }
        return ans;
    }
};