codeforces 1416B. Make Them Equal (#673 Div. 1)

it2023-08-30  65

题目链接:

https://codeforces.com/problemset/problem/1416/B

题目大意:

给你一个数列 长度为 n 在 1e4 之内,每个数字0<ai<1e5,现在你可以执行以下操作

选择一个i,一个j,一个x

使得ai -= i * x   aj += i * x; 

现在问你是否能在3 * n次操作内使得数列中的数字两两相等。

且每次操作完不能出现 ai < 0 的情况。

输入与输出:

输入:

第一行一个 t 代表 t 组数据

对于每组数据:

第一行一个n(1 <= n <= 1e4, t组数据n之和<=1e4)

然后跟着n个数字代表ai

输出:

每组数据第一行一个m代表m次操作

接下来m行每行3个数,代表选定的i, j, k

如果不行则直接输出-1

思路:

首先我们先对数列中的所有数字求一下和,(设为sum)如果sum 除不尽n 则肯定不行

如果可以整除则可以设avr = sum / n

那么我们可以先把所有的数字集中到a1上(因为a1可以随便分配)

那么对于a2 到 an ,我们可以先让ai -> ai + (i - ai % i) % mod 

​这样我们便可以使得 ai / i 能整除,便可以把ai全部转化到a1上。

而且从a2 - > an 的顺序我们便可以保证不会出现负数,因为:

a1一开始至少等于1 而a2转化为可以整除2的a2 又只需要1

同理 这样进行到第ai时,a1至少为 i - 1,便可以放心的把ai变为可整除的

集中到a1之后在逐个分配即可

代码:

#include <cstdio> #include <cstring> #include <algorithm> using namespace std; const int maxn = 1e4 + 5; typedef long long LL; struct node { LL xx; }; struct op { int ii, jj, xx; inline void set(int ii, int jj, int xx){ this->ii = ii; this->jj = jj; this->xx = xx; } op() {} }; node ss[maxn]; op ops[maxn << 2]; int n, opID; int main() { int t; scanf("%d", &t); while (t--) { LL sum = 0; opID = 0; scanf("%d", &n); for (int i = 1; i <= n; i++) { scanf("%lld", &ss[i].xx); sum += ss[i].xx; } if (sum % n) { printf("-1\n"); continue; } else { int avr = sum / n; for (int i = 2; i <= n; i++) { int ned = (i - ss[i].xx % i) % i; ops[++opID].set(1, i, ned); ops[++opID].set(i, 1, (ned + ss[i].xx) / i); } for (int i = 2; i <= n; i++) { ops[++opID].set(1, i, avr); } printf("%d\n", opID); for (int i = 1; i <= opID; i++) printf("%d %d %d\n", ops[i].ii, ops[i].jj, ops[i].xx); } } return 0; }

 

最新回复(0)