結果

問題 No.3062 Rotate and Maximize
ユーザー pockyny
提出日時 2025-07-13 02:52:12
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 51 ms / 2,000 ms
コード長 1,346 bytes
コンパイル時間 787 ms
コンパイル使用メモリ 79,220 KB
実行使用メモリ 7,844 KB
最終ジャッジ日時 2025-07-13 02:52:17
合計ジャッジ時間 3,698 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 51
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <atcoder/modint>
#include <vector>

using namespace std;
using namespace atcoder;
using mint = modint998244353;
int a[3010],b[3010],mx[3010];
bool used[3010];
int main(){
    int i,j = -1,n; cin >> n;
    for(i=0;i<n;i++) cin >> a[i];
    for(i=0;i<n;i++){
        if(a[i]==n){
            j = i;
            break;
        }
    }
    if(j==-1){
        cout << 0 << "\n";
        return 0;
    }
    for(i=0;i<n;i++) b[i] = a[(j + i)%n];
    for(i=0;i<n;i++) a[i] = b[i];
    vector<int> v;
    for(i=0;i<n;i++){
        if(a[i]==n) v.push_back(i);
    }
    for(i=0;i<n;i++){
        int mn = n + 1;
        for(int x:v) mn = min(mn,a[(i + x)%n]);
        mx[i] = mn;
    }
    mint ans = 1;
    used[0] = true;
    for(i=n - 1;i>=1;i--){
        vector<int> cnt(n);
        int c = 0;
        for(j=0;j<n;j++){
            if(a[j]==i){
                c++;
                for(int x:v) cnt[(j - x + n)%n]++;
            }
        }
        int y = -1;
        mint x = 0;
        for(j=0;j<n;j++){
            if(!used[j] && cnt[j]==c && mx[j]>=i){
                x++;
                if(y==-1) y = j;
            }
        }
        if(y==-1){
            ans = 0;
            break;
        }else{
            ans *= x;
            used[y] = true;
        }
    }
    ans *= n;
    cout << ans.val() << "\n";
}
0