这题输出格式没说清楚,好心好意没输出末尾空格,然后就Presentation error,然后加上之后就AC了(黑人问号脸)
思路:
答案只需输出满足消耗最小的列的编号,那我们就只需处理每一列能通过的最小消耗,这当然是一行一行的处理,每行叠加起来就是总的,当存在某一行不能通过此列时,那这一整列都没法通过。
接下来只需思考出怎样在某一行里面计算出某一位置能通过的最小消耗
1.该位置本身没有架子,不需要移动架子,那消耗就是0
2.该位置有架子,那么就要将该架子往左移动或往右移动每行 n 个数,设 s[i] 为第 i 个架子(也可能是空的,空的距离也算1)的末尾的位置, zero[i] 为第 i 个架子(也可能是空的) 包括第 i 个之前的空位的数量。
设占在第 pos 列的架子的宽度范围为 l ~ r,该架子下标为 tt。
那么该架子往左边移动所需的空间为 r - pos + 1,往右边移动所需的空间为 pos - l + 1。
以计算向左移动该架子为例:
需要找到最大的下标 x 使得 zero[tt] - zero[x - 1] >= need_l,这样才能使得消耗: tt - x + 1 - need_l 最小(减need_l是因为空位不用搬动),要找到 x 可用二分,向右移动架子也同理。当然也可能出现向某个方向移动空位不够的情况,那就不能向这个方向移动,如果两个方向都不能移动,那这一列就不能通过了。
至于时间复杂度,每一行是 ni * log(ni),感觉挺玄学的,因为给了5000ms,所以试了一手,最后跑了1660ms
AC代码:
#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>
#include <string>
#include <queue>
#include <map>
#include <stack>
#include <map>
//#include <unordered_map>
#include <vector>
#include <cmath>
//#include <ext/rope>
#include <set>
using namespace std;
//using namespace __gnu_cxx;
#define pair(a, b) make_pair(a, b)
#define memset(a, b) memset(a, b, sizeof a)
#define max(a, b) ((a) < (b) ? (b) : (a))
#define min(a, b) ((a) < (b) ? (a) : (b))
#define fi first
#define se second
int dx[4] = {
0, 1, 0, -1};
int dy[4] = {
1, 0, -1, 0};
//typedef __int128 INT;
typedef long long LL;
typedef unsigned long long ULL;
typedef pair<int, int> PII;
typedef pair<LL, LL> PLL;
const int N = 1e6 + 10;
const int M = 2e7 + 10;
const int Mod = 1e9 + 7;
const int inf = 0x3f3f3f3f;
const LL INF = 0x3f3f3f3f3f3f3f3f;
const int P = 13331;
const double eps = 1e-9;
const double PI = acos(-1.0);
int n, m, k;
int R, L;
int a[M];
LL ans[N], s[M], zero[M];
void work()
{
scanf("%d", &n);
for (int i = 1; i <= n; i ++)
scanf("%d", &a[i]);
s[0] = zero[0] = 0;
for (int i = 1; i <= n; i ++)
{
s[i] = (a[i] ? a[i] : 1);
zero[i] = (!a[i]);
s[i] += s[i - 1];
zero[i] += zero[i - 1];
}
int tt = 1;
for (int i = 1; i <= L; i ++)
{
if (s[tt] < i) tt ++;
if (ans[i] == INF || !a[tt]) continue;
int l = s[tt] - a[tt] + 1, r = s[tt];
int need_l = r - i + 1, need_r = i - l + 1;
int need = inf;
int low = 0, high = tt - 1;
while (low < high)
{
int mid = low + high + 1 >> 1;
if (zero[tt] - zero[mid - 1] >= need_l) low = mid;
else high = mid - 1;
}
if (low >= 1) need = min(need, tt - low + 1 - need_l);
low = tt + 1, high = n + 1;
while (low < high)
{
int mid = low + high >> 1;
if (zero[mid] - zero[tt] >= need_r) high = mid;
else low = mid + 1;
}
if (high <= n) need = min(need, high - tt + 1 - need_r);
if (need == inf) ans[i] = INF;
else ans[i] += need;
}
}
int main()
{
int T;
scanf("%d", &T);
while (T --)
{
memset(ans, 0);
scanf("%d%d", &R, &L);
for (int i = 0; i < R; i ++)
work();
LL res = INF;
for (int i = 1; i <= L; i ++)
res = min(res, ans[i]);
printf("%lld\n", res);
bool flag = false;
for (int i = 1; i <= L; i ++)
if (ans[i] == res)
printf("%d ", i - 1);
printf("\n");
}
return 0;
}