C. Joyboard
这道题需要进行分类讨论。
- 当\(k=1\)时，即构造的数组中所有元素皆为\(0\)才成立，所以输出\(1\)。
- 当\(k=2\)时，只有\(a[n+1]<=n\)或\(a[n + 1]=x\)(其中\(n|x\))才成立，所以答案是\(n+\lfloor \frac{n+m}{n} \rfloor\)\((m>n)\)。
- 当\(k=3\)时，只有\(a[n+1]>n\)且\(a[n + 1]\neq x\)时才成立，所以答案是\(m-n-\lfloor \frac{n+m}{n} \rfloor\)\((m>n)\)。
代码
#include <bits/stdc++.h>
#define endl '\n'
using namespace std;
typedef long long ll;
typedef pair<ll, ll> PII;
const ll N = 1e5 + 10;
ll t;
ll n, m, k;
signed main()
{
std::ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
cin >> t;
while(t--)
{
cin >> n >> m >> k;
if(k > 3)
{
cout << 0 << endl;
}
else if(k == 3)
{
if(m > n)
{
cout << m - n - (m - n) / n << endl;
}
else
{
cout << 0 << endl;
}
}
else if(k == 2)
{
if(m > n)
{
cout << n + (m - n) / n << endl;
}
else
{
cout << m << endl;
}
}
else if(k == 1)
{
cout << 1 << endl;
}
}
return 0;
}