結果

問題 No.1549 [Cherry 2nd Tune] BANning Tuple
ユーザー 👑 chocoruskchocorusk
提出日時 2021-06-11 22:51:37
言語 C++17(1z)
(gcc 10.2.0 + boost 1.73.0)
結果
AC  
実行時間 8 ms / 4,000 ms
コード長 3,132 Byte
コンパイル時間 3,234 ms
使用メモリ 4,512 KB
最終ジャッジ日時 2021-06-11 22:51:42
合計ジャッジ時間 4,679 ms
ジャッジサーバーID
(参考情報)
judge6 / judge11
このコードへのチャレンジ(β)

テストケース

テストケース表示
入力 結果 実行時間
使用メモリ
testcase_00 AC 2 ms
3,408 KB
testcase_01 AC 2 ms
3,560 KB
testcase_02 AC 8 ms
3,568 KB
testcase_03 AC 7 ms
3,876 KB
testcase_04 AC 7 ms
3,796 KB
testcase_05 AC 8 ms
3,796 KB
testcase_06 AC 8 ms
3,732 KB
testcase_07 AC 6 ms
4,468 KB
testcase_08 AC 6 ms
4,344 KB
testcase_09 AC 6 ms
4,432 KB
testcase_10 AC 7 ms
4,344 KB
testcase_11 AC 6 ms
4,460 KB
testcase_12 AC 7 ms
4,372 KB
testcase_13 AC 7 ms
4,512 KB
testcase_14 AC 7 ms
4,372 KB
testcase_15 AC 8 ms
4,396 KB
testcase_16 AC 7 ms
4,368 KB
testcase_17 AC 7 ms
4,340 KB
testcase_18 AC 7 ms
4,456 KB
testcase_19 AC 8 ms
3,472 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <cstdio>
#include <cstring>
#include <iostream>
#include <string>
#include <cmath>
#include <bitset>
#include <vector>
#include <map>
#include <set>
#include <queue>
#include <deque>
#include <algorithm>
#include <complex>
#include <unordered_map>
#include <unordered_set>
#include <random>
#include <cassert>
#include <fstream>
#include <utility>
#include <functional>
#include <time.h>
#include <stack>
#include <array>
#include <list>
#include <atcoder/all>
#define popcount __builtin_popcount
using namespace std;
using namespace atcoder;
typedef long long ll;
typedef pair<int, int> P;

int main()
{
    ll n; int Q; cin>>n>>Q;
    using mint=modint998244353;
    const int mx=3002;
    vector<mint> v(mx+1);
    v[0]=1;
    for(int i=1; i<=mx; i++) v[i]=v[i-1]*mint(n-1+i)/mint(i);
    int i0=0;
    map<ll, vector<int>> mp;
    for(int i=0; i<Q; i++){
        ll k;
        int a, b, s, t;
        cin>>k>>a>>b>>s>>t;
        b++;
        if(mp.find(k)!=mp.end()){
            auto ws=mp[k];
            vector<int> w;
            int q=0;
            for(int j=0; j<=mx; j++){
                if(!q && ws[j]){
                    w.push_back(j);
                    q=1;
                }else if(q && ws[j]==0){
                    w.push_back(j);
                    q=0;
                }
            }
            if(w[0]==0){
                i0-=w[1];
                for(int j=0; j<=mx; j++){
                    for(int m=2; m<w.size(); m++){
                        if(!(m&1) && j>=w[m]-w[1]) v[j]+=v[j-w[m]+w[1]];
                        else if((m&1) && j>=w[m]-w[1]) v[j]-=v[j-w[m]+w[1]];
                    }
                }
            }else{
                for(int j=0; j<=mx; j++){
                    for(int m=0; m<w.size(); m++){
                        if(!(m&1) && j>=w[m]) v[j]+=v[j-w[m]];
                        else if((m&1) && j>=w[m]) v[j]-=v[j-w[m]];
                    }
                }
            }
        }else{
            mp[k].resize(3001);
        }
        for(int j=a; j<b; j++) mp[k][j]++;
        auto ws=mp[k];
        vector<int> w;
        int q=0;
        for(int j=0; j<=mx; j++){
                if(!q && ws[j]){
                    w.push_back(j);
                    q=1;
                }else if(q && ws[j]==0){
                    w.push_back(j);
                    q=0;
                }
        }
        if(w[0]==0){
                i0+=w[1];
                for(int j=mx; j>=0; j--){
                    for(int m=2; m<w.size(); m++){
                        if(!(m&1) && j+w[m]-w[1]<=mx) v[j+w[m]-w[1]]-=v[j];
                        else if((m&1) && j+w[m]-w[1]<=mx) v[j+w[m]-w[1]]+=v[j];
                    }
                }
        }else{
                for(int j=mx; j>=0; j--){
                    for(int m=0; m<w.size(); m++){
                        if(!(m&1) && j+w[m]<=mx) v[j+w[m]]-=v[j];
                        else if((m&1) && j+w[m]<=mx) v[j+w[m]]+=v[j];
                    }
                }
        }
        mint ans=0;
        for(int j=max(0, s-i0); j<=t-i0; j++) ans+=v[j];
        cout<<ans.val()<<endl;
    }
    return 0;
}
0