Codeforces Round 873 (Div. 2)

Divisible Array

题意：

给一个n，要求创建一个数组，使a[i] % i == 0 且数组和sum % n == 0;

思路：

令a[i] = 2 * i即可。

代码：

/*************************************************************************
  > File Name: a.cpp
  > Author: Beans
  > Mail: 3112748286@qq.com 
  > Created Time: 2023/5/15 10:13:25
 ************************************************************************/

#include <iostream>
#define endl '\n'

using namespace std;

int t, n;

signed main(){

    cin >> t;
    while(t -- ){

        cin >> n;
        for(int i = 1; i <= n; i ++ )   cout << i * 2 << ' ';
        cout << endl;
    }
}

Permutation Swap

题意：

给一个排列，a[i] != i，要求找到一个最大的k，你可以交换a[i]和a[i + k],使得a[i] == i。

思路：

找到每一个数的交换长度abs(a[i] - i)，求gcd即可。

代码：

/*************************************************************************
  > File Name: b.cpp
  > Author: Beans
  > Mail: 3112748286@qq.com 
  > Created Time: 2023/5/15 10:16:16
 ************************************************************************/

#include <iostream>
#include <algorithm>
#define endl '\n'

using namespace std;

const int maxn = 3e5 + 7;

int t, n, a[maxn];

int gcd(int a, int b){

    return b ? gcd(b, a % b) : a;
}

void solve(){

    cin >> n;
    for(int i = 1; i <= n; i ++ )
        cin >> a[i];
    int ans = abs(a[1] - 1);
    for(int i = 2; i <= n; i ++ )
        ans = gcd(ans, abs(a[i] - i));
    cout << ans << endl;
}

signed main(){

    ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);
    cin >> t;
    while(t -- )
        solve();
}

Counting Orders

题意：

给出两个数组a和b，求a数组有多少中排列使得a[i]>b[i]，输出排序方案数mod (1e9 + 7)。

思路：

先对a和b按从小到大排序。
对于a[i],在b中找到第一个大于等于它的数，前面的小于它的位置都可以放a[i], 但是得减去前面已经放了数的位置。
即把所有位置的方案数都求出来，相乘即可。

代码：

/*************************************************************************
  > File Name: c.cpp
  > Author: Beans
  > Mail: 3112748286@qq.com 
  > Created Time: 2023/5/15 10:25:43
 ************************************************************************/

#include <iostream>
#include <algorithm>
#define int long long
#define endl '\n'

using namespace std;

const int maxn = 3e5 + 7;

int t, n, a[maxn], b[maxn];

void solve(){

    cin >> n;
    for(int i = 1; i <= n; i ++ )   cin >> a[i];
    for(int i = 1; i <= n; i ++ )   cin >> b[i];
    int ans = 1;
    int mod = 1e9 + 7;
    int pre = 0;
    sort(a + 1, a + n + 1);
    sort(b + 1, b + n + 1);
    for(int i = 1; i <= n; i ++ ){

        ans = (ans * (lower_bound(b + 1, b + n + 1, a[i]) - b - 1 - pre)) % mod;
        pre ++ ;
    }
    cout << ans << endl;
}

signed main(){

    ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);
    cin >> t;
    while(t -- )
        solve();
}

Range Sorting (Easy Version)

题意：

给出一个数组，对所有子数组，找到其中的非升序区间r，l。其花费为r - l。求数组a的总花费。

思路：

先初始化答案，即对于每个i，ans += (i - 1) * ((i - 1) + 1) / 2。
即等差数列求和，把所有合法不合法的花费都加起来，后面再把合法的减去。合法即升序，不合法为非升序。
随后对于每个位置i，找到它前面的合法区间和后面的合法区间，ans减去前面的合法区间*后面的合法区间。
具体看代码。

代码：

/*************************************************************************
  > File Name: d1.cpp
  > Author: Beans
  > Mail: 3112748286@qq.com 
  > Created Time: 2023/5/15 10:48:36
 ************************************************************************/

#include <iostream>
#include <algorithm>
#include <vector>
#define int long long
#define endl '\n'

using namespace std;

const int maxn = 5e3 + 7;

int t, n, a[maxn];

void solve(){

    cin >> n;
    int ans = 0;
    for(int i = 1; i <= n; i ++ )   cin >> a[i], ans += (i - 1) * ((i - 1) + 1) / 2;
    for(int i = 1; i <= n; i ++ ){

        int k = i;
        while(k >= 1 && a[k] >= a[i]) k -- ;
        int x = k;
        if(k >= 1) while(x >= 1 && a[x] <= a[i]) x -- ;
        int y = i;
        while(y <= n && a[y] >= a[i]) y ++ ;
        ans -= (k - x) * (y - i);
    }
    cout << ans << endl;
}

signed main(){

    ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);
    cin >> t;
    while(t -- )
        solve();
}

Range Sorting (Hard Version)

题意：

给出一个数组，对所有子数组，找到其中的非升序区间r，l。其花费为r - l。求数组a的总花费。
hard范围变成了3e5，不能像easy题那样暴力了。

思路：

先初始化答案，即对于每个i，ans += (i - 1) * ((i - 1) + 1) / 2。
即等差数列求和，把所有合法不合法的花费都加起来，后面再把合法的减去。合法即升序，不合法为非升序。
随后对于每个位置i，找到它前面的合法区间和后面的合法区间，ans减去前面的合法区间*后面的合法区间。
这题思路还是一样，但是找合法区间得换一种方法，因为暴力会超时。
至于每个位置后面的合法区间，可以用r数组代替，可以用单调栈的方法先预处理好r数组。
对于前面的合法区间可以用单调栈先找到区间的右端点，然后再用二进制倍增的思想找到左断点。
其中tmi[k][i]表示以i为右端点，长度为2^k的区间的最大值。
具体操作看代码。

代码：

/*************************************************************************
  > File Name: d2.cpp
  > Author: Beans
  > Mail: 3112748286@qq.com 
  > Created Time: 2023/5/15 11:38:39
 ************************************************************************/

#include <iostream>
#include <algorithm>
#include <stack>
#include <vector>
#define int long long
#define endl '\n'

using namespace std;

const int maxn = 3e5 + 7;
const int ML = 19;

int t, n, a[maxn];

void solve(){

    cin >> n; a[n + 1] = 0;
    int ans = 0;
    vector<vector<int>> tmi(ML, vector<int>(n + 1));
    for(int i = 1; i <= n; i ++ )   cin >> a[i], ans += (i - 1) * ((i - 1) + 1) / 2, tmi[0][i] = a[i];
    vector<int> s;
    s.push_back(n + 1);
    vector<int> r(n + 1);
    for(int i = n; i >= 1; i -- ){

        while(s.size() && a[s.back()] > a[i])
            s.pop_back();
        r[i] = s.back();
        s.push_back(i);
    }
    for(int k = 1; k < ML; k ++ )
        for(int i = 1 << k; i <= n; i ++ )
            tmi[k][i] = max(tmi[k - 1][i], tmi[k - 1][i - (1 << (k - 1))]);
    s.clear();
    s.push_back(0);
    for(int i = 1; i <= n; i ++ ){

        while(s.size() && a[s.back()] > a[i])
            s.pop_back();
        int j = s.back();
        for(int k = ML - 1; k >= 0; k -- )
            if(j >= (1 << k) && tmi[k][j] < a[i])
                j -= (1 << k);
        ans -= (r[i] - i) * (s.back() - j);
        s.push_back(i);
    }
    cout << ans << endl;
}

signed main(){

    ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);
    cin >> t;
    while(t -- )
        solve();
}