結果

問題 No.3062 Rotate and Maximize
ユーザー askr58
提出日時 2025-03-14 22:52:26
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 1,549 ms / 2,000 ms
コード長 1,128 bytes
コンパイル時間 6,866 ms
コンパイル使用メモリ 332,616 KB
実行使用メモリ 7,328 KB
最終ジャッジ日時 2025-03-14 22:53:33
合計ジャッジ時間 64,019 ms
ジャッジサーバーID
(参考情報)
judge1 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 51
権限があれば一括ダウンロードができます

ソースコード

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++){
            if(mark[0]==1){
                mark>>=1;
                mark[n-1]=1;
            }else{
                mark>>=1;
            }
            if((mark&now)==mark&&(mark&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