886E - Maximum Element - CodeForces Solution


combinatorics dp math *2400

Please click on ads to support us..

C++ Code:

#include <bits/stdc++.h>

using namespace std;
const int P = 1e9 + 7, N = 1e6;

int qpow(int a, int b)
{
    int res = 1;
    for (; b; b >>= 1, a = 1ll * a * a % P)
        if (b & 1) res = 1ll * res * a % P;
    return res;
}

int n, k, ans, fac, sum;
int g[N + 5];

int main()
{
    scanf("%d%d", &n, &k);
    g[0] = fac = 1, ans = n;
    for (int i = 1; i <= n; i++)
    {
        sum = (sum + g[i - 1]) % P;
        if (i > k) sum = (sum - g[i - k - 1] + P) % P;
        g[i] = 1ll * sum * qpow(i, P - 2) % P;
        ans = (ans - g[i - 1] + P) % P;
        if (i < n) fac = 1ll * fac * i % P; 
    }
    printf("%d\n", 1ll * fac * ans % P);
}


Comments

Submit
0 Comments
More Questions

967. Numbers With Same Consecutive Differences
957. Prison Cells After N Days
946. Validate Stack Sequences
921. Minimum Add to Make Parentheses Valid
881. Boats to Save People
497. Random Point in Non-overlapping Rectangles
528. Random Pick with Weight
470. Implement Rand10() Using Rand7()
866. Prime Palindrome
1516A - Tit for Tat
622. Design Circular Queue
814. Binary Tree Pruning
791. Custom Sort String
787. Cheapest Flights Within K Stops
779. K-th Symbol in Grammar
701. Insert into a Binary Search Tree
429. N-ary Tree Level Order Traversal
739. Daily Temperatures
647. Palindromic Substrings
583. Delete Operation for Two Strings
518. Coin Change 2
516. Longest Palindromic Subsequence
468. Validate IP Address
450. Delete Node in a BST
445. Add Two Numbers II
442. Find All Duplicates in an Array
437. Path Sum III
436. Find Right Interval
435. Non-overlapping Intervals
406. Queue Reconstruction by Height