Submission #68602610

---

Source Code Expand

#include <bits/stdc++.h>

using namespace std;

#ifdef LOCAL
#define DEBUG(...) debug(#__VA_ARGS__, __VA_ARGS__)
#else
#define DEBUG(...) 6
#endif
 
template<typename T, typename S> ostream& operator << (ostream &os, const pair<T, S> &p) {return os << "(" << p.first << ", " << p.second << ")";}
template<typename C, typename T = decay<decltype(*begin(declval<C>()))>, typename enable_if<!is_same<C, string>::value>::type* = nullptr>
ostream& operator << (ostream &os, const C &c) {bool f = true; os << "["; for (const auto &x : c) {if (!f) os << ", "; f = false; os << x;} return os << "]";}
template<typename T> void debug(string s, T x) {cerr << "\033[1;35m" << s << "\033[0;32m = \033[33m" << x << "\033[0m\n";}
template<typename T, typename... Args> void debug(string s, T x, Args... args) {for (int i=0, b=0; i<(int)s.size(); i++) if (s[i] == '(' || s[i] == '{') b++; else
if (s[i] == ')' || s[i] == '}') b--; else if (s[i] == ',' && b == 0) {cerr << "\033[1;35m" << s.substr(0, i) << "\033[0;32m = \033[33m" << x << "\033[31m | "; debug(s.substr(s.find_first_not_of(' ', i + 1)), args...); break;}}

#define int long long
#define pi pair<int,int>
#define mp make_pair
#define pb push_back
#define vi vector<int>
#define eb emplace_back
#define f first
#define s second
#define lep(i,a,b) for (int i = (a); i <= (b); i++)
#define rep(i,a,b) for (int i = (a); i >= (b); i--)
const int inf = 1e18;
int t;

signed main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    int n, k;
    cin >> n >> k;
    int N = n * k;
    vector<int> a(N+1);
    lep(i,1,N) {
        cin >> a[i];
    }
    vector<vector<int>> cycles;
    vector<bool> visited(N+1);
    lep(i,1,N) {
        if (!visited[i]) {
            vector<int> cur_cycle;
            int cur = i;
            while (!visited[cur]) {
                cur_cycle.pb(cur);
                visited[cur] = true;
                cur = a[cur];
            }
            DEBUG(cur_cycle);
            cycles.pb(cur_cycle);
        }
    }
    DEBUG(cycles);

    auto solve = [&](vector<int> cycle) -> int {
        vector<int> pos(N+1,-1);
        int sz = cycle.size();
        lep(i,0,sz-1) {
            pos[cycle[i]] = i;
        }
        vector<vector<int>> dp(sz, vector<int>(sz));
        rep(i,sz-1,0) {
            lep(j,i,sz-1) {
                if (j == i) {
                    dp[i][j] = 0;
                } else {
                    int dp_val = 0;
                    int md = cycle[i] % n;
                    for (int k = md; k <= N; k += n) {
                        if (k == cycle[i]) continue;
                        if (pos[k] >= i and pos[k] <= j) {
                            DEBUG(i,a[i],k,pos[k]);
                            int val = 1;
                            if (pos[k] >= i + 2) {
                                val += dp[i+1][pos[k]-1];
                            }
                            val += dp[pos[k]][j];
                            dp_val = max(dp_val, val);
                        }
                    }
                    dp_val = max(dp_val, dp[i+1][j]);
                    dp_val = max(dp_val, dp[i][j-1]);
                    DEBUG(i,j,dp_val);
                    dp[i][j] = dp_val;
                }
            }
        }
        DEBUG(dp[0][sz-1]);
        return dp[0][sz-1];
    };
    int ans = 0;
    for (auto &x: cycles) {
        ans += solve(x);
    }
    cout << ans << "\n";
}

Submission Info

Compile Error

Main.cpp: In function ‘int main()’:
Main.cpp:8:20: warning: statement has no effect [-Wunused-value]
    8 | #define DEBUG(...) 6
      |                    ^
Main.cpp:52:13: note: in expansion of macro ‘DEBUG’
   52 |             DEBUG(cur_cycle);
      |             ^~~~~
Main.cpp:8:20: warning: statement has no effect [-Wunused-value]
    8 | #define DEBUG(...) 6
      |                    ^
Main.cpp:56:5: note: in expansion of macro ‘DEBUG’
   56 |     DEBUG(cycles);
      |     ^~~~~
Main.cpp: In lambda function:
Main.cpp:8:20: warning: statement has no effect [-Wunused-value]
    8 | #define DEBUG(...) 6
      |                    ^
Main.cpp:75:29: note: in expansion of macro ‘DEBUG’
   75 |                             DEBUG(i,a[i],k,pos[k]);
      |                             ^~~~~
Main.cpp:8:20: warning: statement has no effect [-Wunused-value]
    8 | #define DEBUG(...) 6
      |                    ^
Main.cpp:86:21: note: in expansion of macro ‘DEBUG’
   86 |                     DEBUG(i,j,dp_val);
      |                     ^~~~~
Main.cpp:8:20: warning: statement has no effect [-Wunused-value]
    8 | #define DEBUG(...) 6
      |                    ^
Main.cpp:91:9: note: in expansion of macro ‘DEBUG’
   91 |         DEBUG(dp[0][sz-1]);
      |         ^~~~~

Judge Result