https://codeforces.com/contest/1993/problem/C
乍一眼看上去这题仿佛略有一点想法,之前我就思考过类似的问题,毋庸置疑先从小到大排序,然后根据k来进行周期查看是否所有的后面的灯在点亮的时候都能和前面的接轨同时亮着,这里我们可以对所有元素都进行一个减首个元素(排序后的元素,也就是最小的那个元素)的操作,这样就相当于在0时刻灯亮了,然后在k-1的时刻是灯亮的最后一刻,在k—2k-1又是暗的时刻,这是第一个周期,我们可以根据周期规律对于所有的元素进行对2k求余的操作,这样用一个新的数组b来存储他们求余后的值(这个值相当于在0时刻灯亮的时候,相对与0的值,这个元素亮灯的时刻)。这些值的范围在0—2k-1,在k—2k-1的范围的相当于dui首个元素也就是0时刻而言0已经灭了这个值的位置才开始亮灯,那么要满足所有灯能在同一时刻亮的条件既是两种情况,一种是所有对2k求余后的值都落在0—k-1之间,毋庸置疑全部都能在0时刻没灭的时候亮起来,这样直接思考最接近k-1时刻亮的那盏灯和a(n-1)求余后的值(也就是a[n-1]亮的时刻)的差距,该段距离加上a[n-1]的值就是ans。还有一种情况就是求余后的值落在了k—2k-1的范围,那么该范围是有可能在灯量后持续到0—k-1的一个区间内才灭的,所以我们只需要求出两个值,相当于是判断,一个是在k—2k-1内最早出现的值,这样它灯灭的时候在(0—k-1)范围内灭的时候最早,只有在该范围内才有可能全部的灯一起亮,再判断在它灯灭的时刻(假设是x,那么x的范围是(0-k-2)),这里的范围问题没想清楚可以多举几个例子,在所有对2k求余后的在0—k-1的范围内的起点的值都必须满足也在0—x的范围内,这样才能保证在它灯灭前所有的灯能够一起亮。这就是大概的思路,但是其中还有一些细节,体现在代码中了,读者可以仔细阅读理解意思。
#include <iostream>
#include <vector>
#include <unordered_map>
#include <algorithm>
#include<stack>
using namespace std;
void solve() {
int n, k;
cin >> n >> k;
vector<int>a(n);
vector<int>b(2 * k, 0);
for (int i = 0; i < n; i++)
cin >> a[i];
int ans = 0;
sort(a.begin(), a.end());
/*for(int i=0;i<n-1;i++)
for(int j=i+1;j<n;j++)
if ((a[j] - a[i]) % k == 0 && ((a[j] - a[i]) / k) % 2 == 1)
{
ans = -1;
cout << ans << endl;
return;
}*/
for (int i = 0; i < n; i++)
{
b[(a[i] - a[0]) % (2 * k)]++;
}
int p = 0;
for(int i=k;i<2*k;i++)
if (b[i] > 0)
{
p = i;
break;
}
if (p == 0)
{
int later=0;
for(int i=k-1;i>=0;i--)
if (b[i] > 0)
{
later = i;
break;
}
ans = later - ((a[n - 1] - a[0]) % (2 * k)) + a[n - 1];
}
else {
int pre = (p + k-1) %(2 * k);
for (int i = 0; i <= k - 1; i++)
{
if ((b[i] > 0 && i > pre)|| (p + k - 1<2*k))
{
ans = -1;
}
}
if(ans!=-1){
int sucks = 0;
for(int i=pre;i>=0;i--)
if (b[i] > 0)
{
sucks = i;
break;
}
int end = (a[n - 1] - a[0]) % (2 * k);
if (end <= pre)
ans = a[n-1]+sucks - end;
else {
ans = a[n - 1] + sucks + 2 * k - end;
}
}
}
cout << ans << endl;
}
int main() {
cin.tie(0)->sync_with_stdio(0);
int t;
cin >> t;
while (t--) {
solve();
}
return 0;
}
这段代码是完整的ac代码,然后下面的注释代码是一个判断结果为-1的条件,但是超时了后面就优化了。
/*for(int i=0;i<n-1;i++)
for(int j=i+1;j<n;j++)
if ((a[j] - a[i]) % k == 0 && ((a[j] - a[i]) / k) % 2 == 1)
{
ans = -1;
cout << ans << endl;
return;
}*/
int pre = (p + k-1) %(2 * k);
for (int i = 0; i <= k - 1; i++)
{
if ((b[i] > 0 && i > pre)|| (p + k - 1<2*k))
{
ans = -1;
}
}
优化后只有两种情况才有无法同时亮灯的情况,||左右两边各为一种情况。不懂的可以在评论区留言,若有大神有更好的方法欢迎留言,欢迎指点指正不足之处,感谢阅读。