結果

問題 No.3062 Rotate and Maximize
ユーザー askr58
提出日時 2025-03-14 22:51:07
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 1,047 bytes
コンパイル時間 5,995 ms
コンパイル使用メモリ 332,792 KB
実行使用メモリ 14,900 KB
最終ジャッジ日時 2025-03-14 22:51:17
合計ジャッジ時間 9,854 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 8 TLE * 1 -- * 42
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#include <atcoder/all>
using namespace std;
using namespace atcoder;
using ll=long long;
using mint=modint998244353;
int main()
{
    int n;
    cin>>n;
    vector<int> a(n);
    for(int i=0;i<n;i++)cin>>a[i],a[i]--;
    vector<vector<int>> idx(n);
    for(int i=0;i<n;i++){
        idx[a[i]].push_back(i);
    }
    if(idx[n-1].size()==0){
        cout<<0<<endl;
        return 0;
    }
    int m=idx[n-1].size();
    bitset<3000> mark;
    for(int x:idx[n-1])mark[x]=1;
    bitset<3000> mask;
    for(int i=0;i<n;i++)mask[i]=1;
    bitset<3000> now=mark;
    mint ans=n;
    for(int i=n-2;i>=0;i--){
        bitset<3000> bs;
        for(int x:idx[i])bs[x]=1;
        now|=bs;
        mint res=0;
        for(int j=0;j<n;j++){
            bitset<3000> bbs=((((mark>>j)|(mark<<(n-j))))&mask);
            if((bbs&now)==bbs&&(bbs&bs)==bs){
                res++;
            }
        }
        if(idx[i].size()==0)ans=ans*(res-(n-i-1));
        else ans=ans*res;
    }
    cout<<ans.val()<<endl;
    
    

    return 0;
}
0