結果

問題 No.1596 Distance Sum in 2D Plane
ユーザー BSerBSer
提出日時 2021-07-09 22:58:53
言語 C++14
(gcc 11.2.0 + boost 1.78.0)
結果
AC  
実行時間 169 ms / 2,000 ms
コード長 3,371 bytes
コンパイル時間 3,532 ms
使用メモリ 15,496 KB
最終ジャッジ日時 2023-02-02 00:57:11
合計ジャッジ時間 7,362 ms
ジャッジサーバーID
（参考情報） 		judge13 / judge14
このコードへのチャレンジ（β）

テストケース

テストケース表示
入力 結果 実行時間
使用メモリ
testcase_00 AC 21 ms
15,376 KB
testcase_01 AC 20 ms
15,372 KB
testcase_02 AC 168 ms
15,380 KB
testcase_03 AC 164 ms
15,424 KB
testcase_04 AC 165 ms
15,372 KB
testcase_05 AC 166 ms
15,440 KB
testcase_06 AC 166 ms
15,472 KB
testcase_07 AC 169 ms
15,440 KB
testcase_08 AC 168 ms
15,436 KB
testcase_09 AC 166 ms
15,476 KB
testcase_10 AC 169 ms
15,424 KB
testcase_11 AC 154 ms
15,376 KB
testcase_12 AC 154 ms
15,472 KB
testcase_13 AC 155 ms
15,472 KB
testcase_14 AC 21 ms
15,440 KB
testcase_15 AC 19 ms
15,444 KB
testcase_16 AC 19 ms
15,468 KB
testcase_17 AC 20 ms
15,428 KB
testcase_18 AC 19 ms
15,476 KB
testcase_19 AC 19 ms
15,496 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#include <atcoder/all>
using namespace std;
using namespace atcoder;
using ll = long long;
struct Edge
{
    ll to;
    ll cost;
};
using Graph = vector<vector<ll>>;
using P = pair<ll, ll>;
#define mp make_pair
#define REP(i, n) for (int i = 0; i < (n); ++i)
#define SORT(v) sort((v).begin(), (v).end())
#define RSORT(v) sort((v).rbegin(), (v).rend())
#define ALL(v) (v).begin(), v.end()
const ll nmax = 8;
const ll INF = 1e10;
const int MAX = 510000;
const ll MOD = 1000000007;
bool graph[nmax][nmax];
long long fac[MAX], finv[MAX], inv[MAX];
void Case(int i) { printf("Case #%d: ", i); }
void COMinit()
{
    fac[0] = fac[1] = 1;
    finv[0] = finv[1] = 1;
    inv[1] = 1;
    for (int i = 2; i < MAX; i++)
    {
        fac[i] = fac[i - 1] * i % MOD;
        inv[i] = MOD - inv[MOD % i] * (MOD / i) % MOD;
        finv[i] = finv[i - 1] * inv[i] % MOD;
    }
}

ll COM(int n, int k)
{
    if (n < k)
        return 0;
    if (n < 0 || k < 0)
        return 0;

    return fac[n] * (finv[k] * finv[n - k] % MOD) % MOD;
}

vector<vector<ll>> dist = vector<vector<ll>>(nmax, vector<ll>(nmax, INF));
void warshall_floyd(ll n)
{
    for (size_t i = 0; i < n; i++)
    {
        for (size_t j = 0; j < n; j++)
        {
            for (size_t k = 0; k < n; k++)
            {
                dist[j][k] = min(dist[j][k], dist[j][i] + dist[i][k]);
            }
        }
    }
}

ll gcd(ll a, ll b)
{
    if (b == 0)
        return a;
    return gcd(b, a % b);
}

ll lcm(ll a, ll b)
{
    ll g = gcd(a, b);
    return a / g * b;
}

ll mulMod(ll a, ll b)
{
    return (((a % MOD) * (b % MOD)) % MOD);
}

ll powMod(ll a, ll p)
{
    if (p == 0)
    {
        return 1;
    }
    else if (p % 2 == 0)
    {
        ll half = powMod(a, p / 2);
        return mulMod(half, half);
    }
    else
    {
        return mulMod(powMod(a, p - 1), a);
    }
}

ll ceil(ll a, ll b)
{
    return (a + b - 1) / b;
}

bool isPrime(ll N)
{
    for (int i = 2; i * i <= N; i++)
    {
        if (N % i == 0)
            return false;
    }
    return true;
}

ll cntbit(ll i)
{
    if (i == 0)
        return 0;

    ll ret = 0;
    i++;
    while (i != 0)
    {
        i = i >> 1;
        ret++;
    }
    return ret - 1;
}

ll getSize(ll a)
{
    ll ret = 0;
    while (a != 0)
    {
        ret++;
        a /= 10;
    }
    return ret;
}

vector<pair<long long, long long>> prime_factorize(long long N)
{
    vector<pair<long long, long long>> res;
    for (long long a = 2; a * a <= N; ++a)
    {
        if (N % a != 0)
            continue;
        long long ex = 0;

        while (N % a == 0)
        {
            ++ex;
            N /= a;
        }

        res.push_back({a, ex});
    }

    if (N != 1)
        res.push_back({N, 1});
    return res;
}

using mint = modint1000000007;

int main()
{
    ll N, M;
    cin >> N >> M;
    mint ans;
    COMinit();

    ans = COM(2 * N, N);
    ans *= 2;
    ans *= N;
    // cout << COM(0,0) << endl;
    for (int i = 0; i < M; i++)
    {
        ll t, x, y;
        cin >> t >> x >> y;
        mint tmp = COM(x + y, min(x, y));
        if (t == 1)
        {
            tmp *= COM(2 * N - x - y - 1, min(N - x - 1, N - y));
        }
        else
        {
            tmp *= COM(2 * N - x - y - 1, min(N - x, N - y - 1));
        }
        ans -= tmp;
    }
    cout << ans.val() << endl;

    return 0;
}
0