結果

問題 No.1111 コード進行
コンテスト
ユーザー jjjjjjjtgpptmjj
提出日時 2020-07-10 22:50:54
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 288 ms / 2,000 ms
コード長 4,080 bytes
コンパイル時間 2,068 ms
コンパイル使用メモリ 204,808 KB
最終ジャッジ日時 2025-01-11 19:03:04
ジャッジサーバーID
（参考情報） 		judge4 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 48
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
#define rep(i,N) for(int i=0;i<int(N);++i)
#define rep1(i,N) for(int i=1;i<int(N);++i)
#define all(a) (a).begin(),(a).end()
#define print(v) { cerr<<#v<<": [ "; for(auto _ : v) cerr<<_<<", "; cerr<<"]"<<endl; }
#define printpair(v) { cerr<<#v<<": [ "; for(auto _ : v) cerr<<"{"<<_.first<<","<<_.second<<"}"<<", "; cerr<<"]"<<endl; }
#define dump(x) cerr<<#x<<": "<<x<<endl;
#define bit(k) (1LL<<(k))
typedef long long ll;
typedef pair<int, int> i_i;
typedef pair<ll, ll> l_l;
template <class T> using vec = vector<T>;
template <class T> using vvec = vector<vec<T>>;
template< typename T1, typename T2 >
ostream &operator<<(ostream &os, const pair< T1, T2 >& p) {
  os << "{" <<p.first << ", " << p.second << "}";
  return os;
}
template<class T> inline bool chmax(T& a, T b) { if (a < b) { a = b; return true; } return false; }
template<class T> inline bool chmin(T& a, T b) { if (a > b) { a = b; return true; } return false; }

const int INF = (ll)1e9;
const ll INFLL = (ll)1e18+1;
const ll MOD = (ll)1e9+7;
const double PI = acos(-1.0);
/*
const int dx[8] = {1, 0, -1, 0, 1, -1, -1, 1};
const int dy[8] = {0, 1, 0, -1, 1, 1, -1, -1};
const string dir = "DRUL";
*/

struct mint {
    long long x;
    mint(long long _x=0):x((_x%MOD+MOD)%MOD){}
    mint operator-() const { return mint(-x);}
    mint& operator+=(const mint a) {
        if ((x += a.x) >= MOD) x -= MOD;
        return *this;
    }
    mint& operator-=(const mint a) {
        if ((x += MOD-a.x) >= MOD) x -= MOD;
        return *this;
    }
    mint& operator*=(const mint a) {
        (x *= a.x) %= MOD;
        return *this;
    }
    mint operator+(const mint a) const {
        mint res(*this);
        return res+=a;
    }
    mint operator-(const mint a) const {
        mint res(*this);
        return res-=a;
    }
    mint operator*(const mint a) const {
        mint res(*this);
        return res*=a;
    }
    mint modpow(long long t) const {
        if (!t) return 1;
        mint a = modpow(t>>1);
        a *= a;
        if (t&1) a *= *this;
        return a;
    }
    // for prime MOD
    mint inv() const {
        return modpow(MOD-2);
    }
    mint& operator/=(const mint a) {
        return (*this) *= a.inv();
    }
    mint operator/(const mint a) const {
        mint res(*this);
        return res/=a;
    }
    friend std::ostream& operator<<(std::ostream& os, const mint& a){
        os << a.x;
        return os;
    }
};
// dp[i][j]:i個決めたときの複雑さjの個数
// ans = dp[N][K];
// その前がなんのコードだったかを持たないといけないからdp[i][j][k]になるか？
// 300 ^ 4になって間に合わない
// コード進行は有向グラフに落ちるので、そこからなんかできないかなー

// bfsで探索しながら、N-1回遷移したら終わる感じにしたらどう？
const int sz = 333;
mint dp[sz][sz][sz];
int main() {
    cin.tie(0);
    ios::sync_with_stdio(false);
    cout << fixed << setprecision(20);

    ll N, M, K;
    cin >> N >> M >> K;
    vvec<l_l> G(sz);
    rep(i,M){
        ll p, q, c;
        cin >> p >> q >> c;
        p--, q--;
        G[p].push_back({q, c});
    }
    rep(i,sz)rep(j,sz)rep(k,sz)dp[i][j][k] = 0;

    mint ans = 0;

    /* calc */
    // {頂点,移動回数,複雑度合計,個数}
    queue<pair<l_l,pair<ll, mint>>> q;
    rep(i,sz)q.push({{i, 1}, {0, 1}});
    rep(i,sz)dp[i][1][0] =1;
    while(not q.empty()){
        auto tmp = q.front();
        q.pop();
        int cur = tmp.first.first;
        int cnt = tmp.first.second;
        ll sum = tmp.second.first;
        mint dp_sum = tmp.second.second;
        if(cnt == N)continue;
        if(dp_sum.x != dp[cur][cnt][sum].x)continue;
        for(auto &p: G[cur]){
            int nxt = p.first;
            int C = p.second;
            if(sum + C > K)continue;
            dp[nxt][cnt+1][sum + C] += dp_sum;
            q.push({{nxt, cnt+1},{sum + C, dp[nxt][cnt+1][sum + C]}});
        }
    }
    rep(i,sz){
        ans += dp[i][N][K];
    }
    cout << ans << endl;
}
0