C. Set or Decrease
题目
知识点
思维
解题思路
该题可以直接 O ( n ) O(n) O(n) 遍历求解,也能使用三分,二分等方法求解。
因为昨天三分写寄了此处说明
O
(
n
)
O(n)
O(n) 遍历求解的方法,我们先对数组进行升序排序,很明显可以知道最小操作次数的方案一定是先对
a
1
a_1
a1 进行
n
u
m
1
num_1
num1 次
1
1
1 操作,然后再对
[
n
−
n
u
m
2
+
1
,
n
]
[n - num_2 + 1, n]
[n−num2+1,n] 的数进行
2
2
2 操作(令
a
i
=
a
1
a_i = a_1
ai=a1),总操作次数等于
n
u
m
1
+
n
u
m
2
num_1 + num_2
num1+num2 。
现在我们去遍历 n u m 2 num_2 num2 , 然后求相应的 n u m 1 num_1 num1 的最小次数,假设我们对 [ n − n u m 2 + 1 , n ] [n - num_2 + 1, n] [n−num2+1,n] 位置的数都进行 2 2 2 操作后仍然需要再减 P P P 才能满足 s u m ≤ k sum ≤ k sum≤k , 由于我们 n u m 1 num_1 num1 每增加 1 1 1, 我们的总数 s u m sum sum 就会减小 n u m + 1 num+1 num+1, 所以我们 n u m 1 num_1 num1 的最小次数就是 ⌈ P n u m 1 + 1 ⌉ \lceil \frac{P}{num_1+1}\rceil ⌈num1+1P⌉ ,然后我们取所有 n u m 1 + n u m 2 num_1+num_2 num1+num2 中的最小值就是答案了。
(详细见代码)
代码
#include <bits/stdc++.h>
#define debug(x) cerr << #x << " = " << x << '\n'
#define between(x, l, r) (x >= l && x <= r)
#define point(a, b, m) ((a - 1) * m + b)
#define all(x) (x).begin(), (x).end()
#define SUM(a) accumulate(all(a), 0LL)
#define MIN(a) (*min_element(all(a)))
#define MAX(a) (*max_element(all(a)))
#define np next_permutation
#define lb lower_bound
#define ub upper_bound
#define eb emplace_back
#define pb push_back
#define y second
#define x first
using namespace std;
typedef long long LL;
typedef pair<int, int> PII;
typedef pair<double, double> PDD;
const int N = 3e5 + 10, M = 2 * N, mod = 1e9 + 7;
LL n, k;
LL a[N], s[N];
void solve()
{
cin >> n >> k;
for (int i = 1; i <= n; i ++ ) cin >> a[i];
sort(a + 1, a + n + 1);
for(int i = 2; i <= n; i ++ ) s[i] = s[i - 1] + a[i];
LL sum = a[1] + s[n];
if(sum <= k)
{
cout << 0 << '\n';
return;
}
sum -= k;
for (int i = n; i >= 2; i -- )
{
LL now = (k - s[i - 1]) / (n - i + 2);
LL x1 = now * (n - i + 2), x2 = k - s[i - 1];
if(x1 > x2) now -- ;
sum = min(sum, max(a[1] - now, 0LL) + (n - i + 1));
if(now > a[1]) break;
}
cout << sum << '\n';
}
int main()
{
ios::sync_with_stdio(0), cin.tie(0);
cout.precision(2), cout.setf(ios::fixed);
int t;
cin >> t;
while (t -- ) solve();
return 0;
}
D. Shuffle
题目
知识点
组合数学
解题思路
我们遍历每个包含 k k k 的最长的子串,例如
7 2
1100110
中就是1100110 , 1100110 , 1100110 ,我们先加上这三个组的方案数 C 4 2 C_4^2 C42 , C 4 2 C_4^2 C42 , C 5 2 C_5^2 C52 , 然后我们观察到相邻组之间我们进行了重复计算比如一二组的 1100110 部分, 二三组的 1100110 的部分,所以我们将重复计算的部分 C 3 1 , C 3 1 C_3^1,C_3^1 C31,C31 减去, 剩下的就是答案了。
现在问题是重复的部分我们该怎么求,我们观察发现,该部分是由于 第 i i i 组中第 1 1 1 个 1 1 1 没改变位置的方案,与第 i + 1 i + 1 i+1 组中最后一个 1 1 1 没有改变位置的方案重叠了,例如第一组 1100 与第二组 1001 , 所以我们只需要减去每一组(除了最后一组)固定第一个 1 1 1 后的方案就行了。
(详细见代码)
代码
#include <bits/stdc++.h>
#define debug(x) cerr << #x << " = " << x << '\n'
#define between(x, l, r) (x >= l && x <= r)
#define point(a, b, m) ((a - 1) * m + b)
#define all(x) (x).begin(), (x).end()
#define SUM(a) accumulate(all(a), 0LL)
#define MIN(a) (*min_element(all(a)))
#define MAX(a) (*max_element(all(a)))
#define np next_permutation
#define lb lower_bound
#define ub upper_bound
#define eb emplace_back
#define pb push_back
#define y second
#define x first
using namespace std;
typedef long long LL;
typedef pair<int, int> PII;
typedef pair<double, double> PDD;
const int N = 3e5 + 10, M = 2 * N, mod = 998244353;
LL inv[N], fac[N], ifac[N];
void init(int n)
{
inv[1] = fac[0] = fac[1] = ifac[0] = ifac[1] = 1;
for (int i = 2; i <= n; i ++ )
{
inv[i] = (mod - mod / i) * inv[mod % i] % mod;
fac[i] = fac[i - 1] * i % mod;
ifac[i] = ifac[i - 1] * inv[i] % mod;
}
}
LL C(int m, int n) { return fac[m] * ifac[n] % mod * ifac[m - n] % mod; }
int pre[N], fix[N]; // 第i位左右连续0最远的位置
string s;
int pos[N]; // 第i个1的位置
void solve()
{
int n, k;
cin >> n >> k >> s;
s = " " + s;
//第i位左右连续0最远的位置
int last = 1;
for (int i = 1; i <= n; i ++ )
{
if(s[i] == '1') pre[i] = last, last = i + 1;
}
last = n;
for (int i = n; i >= 1; i -- )
{
if(s[i] == '1') fix[i] = last, last = i - 1;
}
//第i个1的位置
int cnt = 0;
for (int i = 1; i <= n; i ++ )
if(s[i] == '1') pos[++ cnt] = i;
// 特判
if(cnt < k || k == 0)
{
cout << 1 << '\n';
return;
}
//记录答案
LL res = 0;
for (int i = 1; i <= cnt - k + 1; i ++ )
{
//一个组随便排列的方案数
int L = pre[pos[i]], R = fix[pos[i + k - 1]];
//与下一个组的重复部分
int L1 = pre[pos[i + 1]], R1 = R;
res += C(R - L + 1, k);
if(i != cnt - k + 1) res -= C(R1 - L1 + 1, k - 1);
// cout << i << ' ' << L << ' ' << R << " " << L1 << ' ' << R1 << ' ' << res << '\n';
res = (res + mod) % mod;
}
cout << res << '\n';
}
int main()
{
ios::sync_with_stdio(0), cin.tie(0);
cout.precision(2), cout.setf(ios::fixed);
init(1e4 + 10);
// int t;
// cin >> t;
// while (t -- )
solve();
return 0;
}
E. Math Test
题目
知识点
bitmasks 贪心
解题思路
我们用一个二进制数 k k k 的第 i i i 位 1 1 1 或 0 0 0 来代表第 i i i 个人的 ∣ r i − x i ∣ |r_i - x_i| ∣ri−xi∣ 去掉绝对值后是 r i − x i r_i - x_i ri−xi 还是 − ( r i − x i ) -(r_i - x_i) −(ri−xi) 。
由于 ∣ r i − x i ∣ ≥ 0 |r_i - x_i| ≥ 0 ∣ri−xi∣≥0 , 所以当第 i i i 个人的符号取错时答案只可能减小不可能增加,因此我取最大值正确性是能保证的。
首先,我们遍历 k = [ 0 , 2 n − 1 ] k = [0, 2^n - 1] k=[0,2n−1] , 然后求出每一道题 i i i ,答对的人中有多少人去绝对值后是正的,多少人是负的,假设有 x x x 个人是正的, y y y 个人是负的,那么我第 i i i 题对答案的贡献就是 p i ∗ ( x i − y i ) p_i * (x_i - y_i) pi∗(xi−yi) 。
我们想要最大化 ∑ i = 1 m p i ∗ ( x i − y i ) \sum_{i = 1}^m{p_i * (x_i - y_i)} ∑i=1mpi∗(xi−yi) , 只需要按照 ( x i − y i ) (x_i - y_i) (xi−yi) 的大小升序分配 p i p_i pi 即可,例如 ( x − y ) = [ − 3 , 1 , 0 , 4 , − 2 ] (x - y) = [-3, 1, 0, 4, -2] (x−y)=[−3,1,0,4,−2] , 那么我 p = [ 1 , 4 , 3 , 5 , 2 ] p = [1,4,3,5,2] p=[1,4,3,5,2] 是最大的。
(详细见代码)
代码
#include <bits/stdc++.h>
#define debug(x) cerr << #x << " = " << x << '\n'
#define between(x, l, r) (x >= l && x <= r)
#define point(a, b, m) ((a - 1) * m + b)
#define all(x) (x).begin(), (x).end()
#define SUM(a) accumulate(all(a), 0LL)
#define MIN(a) (*min_element(all(a)))
#define MAX(a) (*max_element(all(a)))
#define np next_permutation
#define lb lower_bound
#define ub upper_bound
#define eb emplace_back
#define pb push_back
#define y second
#define x first
using namespace std;
typedef long long LL;
typedef pair<int, int> PII;
typedef pair<double, double> PDD;
const int N = 3e5 + 10, M = 2 * N, mod = 998244353;
int x[N];
string s[N];
int ans[N];
void solve()
{
int n, m;
cin >> n >> m;
for (int i = 1; i <= n; i ++ ) cin >> x[i];
for (int i = 1; i <= n; i ++ ) cin >> s[i], s[i] = " " + s[i];
LL res = -1, tmp;
for (int k = 0; k < (1 << n); k ++ ) // 第i个绝对值是取正还是负
{
// x_i的总和
tmp = 0;
for (int i = 0; i < n; i ++ )
{
int r = 1;
if(k >> i & 1) r = -1;
tmp += r * x[i + 1];
}
vector<PII> num;
for (int i = 1; i <= m; i ++ ) // 第i位
{
int tmp = 0;
for (int j = 0; j < n; j ++ )
{
// 如果是1, 那就是加上, 否则就是减去
int r = -1;
if(k >> j & 1) r = 1;
tmp += (s[j + 1][i] - '0') * r;
}
num.push_back({tmp, i}); // {x_i - y_i, i}
}
// (x_i - y_i)小的就少分配点
sort(all(num));
for (int i = 1; i <= m; i ++ )
tmp += num[i - 1].x * i;
if(res < tmp)
{
res = tmp;
for (int i = 0; i < m; i ++ )
{
auto &[val, id] = num[i];
ans[id] = i + 1;
}
}
}
// cout << res << '\n';
for (int i = 1; i <= m; i ++ ) cout << ans[i] << ' ';
cout << '\n';
}
int main()
{
ios::sync_with_stdio(0), cin.tie(0);
cout.precision(2), cout.setf(ios::fixed);
int t;
cin >> t;
while (t -- )
solve();
return 0;
}