Submission #1707148


Source Code Expand

Copy
// {{{ by shik
#include <bits/stdc++.h>
#include <unistd.h>
#pragma GCC diagnostic ignored "-Wunused-result"
#pragma GCC diagnostic ignored "-Wunused-function"
#define SZ(x) ((int)(x).size())
#define ALL(x) begin(x), end(x)
#define REP(i, n) for (int i = 0; i < int(n); i++)
#define REP1(i, a, b) for (int i = (a); i <= int(b); i++)
#define MP make_pair
#define PB push_back
using namespace std;
typedef int64_t LL;
typedef pair<int, int> PII;
typedef vector<int> VI;

namespace { namespace shik {

template<class T> void _R(T &x) { cin >> x; }
void _R(int &x) { scanf("%d", &x); }
void _R(int64_t &x) { scanf("%" SCNd64, &x); }
void _R(double &x) { scanf("%lf", &x); }
void _R(char &x) { scanf(" %c", &x); }
void _R(char *x) { scanf("%s", x); }
void R() {}
template<class T, class... U> void R(T &head, U &... tail) { _R(head); R(tail...); }

template<class T> void _W(const T &x) { cout << x; }
void _W(const int &x) { printf("%d", x); }
void _W(const int64_t &x) { printf("%" PRId64, x); }
void _W(const double &x) { printf("%.16f", x); }
void _W(const char &x) { putchar(x); }
void _W(const char *x) { printf("%s", x); }
template<class T> void _W(const vector<T> &x) { for (auto i = x.begin(); i != x.end(); _W(*i++)) if (i != x.cbegin()) putchar(' '); }
void W() {}
template<class T, class... U> void W(const T &head, const U &... tail) { _W(head); putchar(sizeof...(tail) ? ' ' : '\n'); W(tail...); }

#ifdef SHIK
#include "dump.hpp"
#else
#define dump(...)
#endif

template<class T, class F = less<T>> void sort_uniq(vector<T> &v, F f = F()) { sort(begin(v), end(v), f); v.resize(unique(begin(v), end(v)) - begin(v)); }
template<class T> inline T bit(T x, int i) { return (x >> i) & 1; }
template<class T> inline bool chkmax(T &a, const T &b) { return b > a ? a = b, true : false; }
template<class T> inline bool chkmin(T &a, const T &b) { return b < a ? a = b, true : false; }
template<class T> using MaxHeap = priority_queue<T>;
template<class T> using MinHeap = priority_queue<T, vector<T>, greater<T>>;

// }}}

// {{{ ModInt
template<int MOD>
struct ModInt {
    int x;
    ModInt() : x(0) {}
    template<class T, typename enable_if<is_integral<T>::value>::type* = nullptr>
    ModInt(T _x) {
        x = _x % MOD;
        if (x < 0) x += MOD;
    }
    ModInt operator-() const { return {x == 0 ? 0 : MOD-x}; }
    ModInt& operator+=(ModInt rhs) {
        x += rhs.x;
        if (x >= MOD) x -= MOD;
        return *this;
    }
    ModInt& operator-=(ModInt rhs) {
        x -= rhs.x;
        if (x < 0) x += MOD;
        return *this;
    }
    ModInt& operator*=(ModInt rhs) {
        x = (long long)x * rhs.x % MOD;
        return *this;
    }
    ModInt& operator/=(ModInt rhs) {
        return *this *= rhs.inv();
    }
    ModInt operator+(ModInt rhs) const { return ModInt(*this) += rhs; }
    ModInt operator-(ModInt rhs) const { return ModInt(*this) -= rhs; }
    ModInt operator*(ModInt rhs) const { return ModInt(*this) *= rhs; }
    ModInt operator/(ModInt rhs) const { return ModInt(*this) /= rhs; }
    ModInt inv() const {
        // should work for non-prime MOD if gcd(x, MOD) = 1
        int a = x, b = MOD, u = 1, v = 0;
        while (b != 0) {
            int t = a / b;
            a -= t * b;
            u -= t * v;
            swap(a, b);
            swap(u, v);
        }
        return u;
    }
    ModInt pow(long long e) {
        ModInt r = 1, p = *this;
        while (e) {
            if (e & 1) r *= p;
            p *= p;
            e >>= 1;
        }
        return r;
    }
    bool operator==(ModInt rhs) { return x == rhs.x; }
    bool operator!=(ModInt rhs) { return x != rhs.x; }
    bool operator<(ModInt rhs) { return x < rhs.x; }
    bool operator<=(ModInt rhs) { return x <= rhs.x; }
    bool operator>(ModInt rhs) { return x > rhs.x; }
    bool operator>=(ModInt rhs) { return x >= rhs.x; }
    friend ostream& operator<<(ostream &os, ModInt i) { return os << i.x; }
};
// }}}

const int MOD = 1e9 + 7;
using mint = ModInt<MOD>;

const int N=410;
int n,a[N],b[N],cnt[N][N];
bitset<N> has[N][N];
mint dp[N][N][N/3][3];
void main() {
    R(n);
    REP1(i,1,n) R(a[i]);
    REP1(i,1,n) R(b[i]);
    REP1(i,1,n) {
        has[i][0]=has[i-1][0];
        has[i][0][a[i]]=1;
        REP1(j,1,n) {
            has[i][j]=has[i][j-1];
            has[i][j][b[j]]=1;
        }
    }
    REP1(i,1,n) REP1(j,1,n) cnt[i][j]=has[i][j].count();
    dp[1][1][0][0]=1;
    mint ans=0;
    REP1(i,1,n+1) REP1(j,1,n+1) REP1(k,0,n/3) REP(l,3) {
        mint me=dp[i][j][k][l];
        if ( me.x==0 ) continue;
        // dump(i,j,k,me);
        if ( i>n || j>n ) {
            if ( k==0 && l==0 ) ans+=me;
            continue;
        }
        if ( l==0 ) {
            if ( has[i-1][j-1][a[i]] ) dp[i+1][j][k][l]+=me;
            else {
                if ( k>0 ) dp[i+1][j][k-1][l]+=me*k;
                dp[i][j][k][l+1]+=me;
            }
        } else if ( l==1 ) {
            if ( has[i-1][j-1][b[j]] ) dp[i][j+1][k][l]+=me;
            else {
                if ( a[i]!=b[j] && k>0 ) dp[i][j+1][k-1][l]+=me*k;
                dp[i][j][k][l+1]+=me;
            }
        } else if ( l==2 ) {
            if ( a[i]!=b[j] ) dp[i+1][j+1][k+1][0]+=me;
        }
    }
    // dump(ans);
    REP1(i,1,n) if ( i%3!=0 ) ans*=i;
    W(ans);
}

// {{{ main
}}
int main() { shik::main(); return 0; }
// }}}

Submission Info

Submission Time
Task F - Three Gluttons
User shik
Language C++14 (GCC 5.4.1)
Score 1800
Code Size 5515 Byte
Status
Exec Time 206 ms
Memory 277248 KB

Test Cases

Set Name Score / Max Score Test Cases
Sample 0 / 0 0_00.txt, 0_01.txt, 0_02.txt, 0_03.txt, 0_04.txt
All 1800 / 1800 0_00.txt, 0_01.txt, 0_02.txt, 0_03.txt, 0_04.txt, 1_00.txt, 1_01.txt, 1_02.txt, 1_03.txt, 1_04.txt, 1_05.txt, 1_06.txt, 1_07.txt, 1_08.txt, 1_09.txt, 1_10.txt, 1_11.txt, 1_12.txt, 1_13.txt, 1_14.txt, 1_15.txt, 1_16.txt, 1_17.txt, 1_18.txt, 1_19.txt, 1_20.txt, 1_21.txt, 1_22.txt, 1_23.txt, 1_24.txt, 1_25.txt, 1_26.txt, 1_27.txt, 1_28.txt, 1_29.txt, 1_30.txt, 1_31.txt
Case Name Status Exec Time Memory
0_00.txt 110 ms 268544 KB
0_01.txt 108 ms 268544 KB
0_02.txt 110 ms 268544 KB
0_03.txt 109 ms 268544 KB
0_04.txt 110 ms 268544 KB
1_00.txt 179 ms 277120 KB
1_01.txt 199 ms 277120 KB
1_02.txt 198 ms 277120 KB
1_03.txt 200 ms 277120 KB
1_04.txt 202 ms 277120 KB
1_05.txt 194 ms 277120 KB
1_06.txt 196 ms 277120 KB
1_07.txt 203 ms 277120 KB
1_08.txt 205 ms 277120 KB
1_09.txt 206 ms 277120 KB
1_10.txt 203 ms 276992 KB
1_11.txt 203 ms 276992 KB
1_12.txt 198 ms 276864 KB
1_13.txt 206 ms 277248 KB
1_14.txt 205 ms 277120 KB
1_15.txt 206 ms 277120 KB
1_16.txt 202 ms 277120 KB
1_17.txt 202 ms 277120 KB
1_18.txt 198 ms 277120 KB
1_19.txt 197 ms 277120 KB
1_20.txt 198 ms 276992 KB
1_21.txt 202 ms 277120 KB
1_22.txt 202 ms 277120 KB
1_23.txt 202 ms 277120 KB
1_24.txt 203 ms 277120 KB
1_25.txt 202 ms 277120 KB
1_26.txt 199 ms 276864 KB
1_27.txt 204 ms 277120 KB
1_28.txt 201 ms 277120 KB
1_29.txt 204 ms 277120 KB
1_30.txt 200 ms 276992 KB
1_31.txt 197 ms 276864 KB