题意:
有一个大小为n的数组,只对每对相邻的两个元素x,y 进行三种操作之一:
1).x给y一个数
2).y给x一个数
3).不变(不操作)
结果要让所有的数都相等,输出1,2操作的个数,并且给出具体的操作。思路:
就两个点,数组总和是否能够整除n,如果不能就肯定不能满足要求;
另外一点就是按照题目说的模拟操作。由于任意相邻的元素都只能进行三个操作中的一种,所以可以从左到右边进行遍历,从2开始遍历三次(个人从1开始存数据)。
第一次,不对第一个元素做任何操作,遍历;
第二次,第一个元素给第二个元素一个数,遍历;
第三次,第二个元素给第一个元素一个数,遍历;
遍历如果存在一个能够满足条件的就按要求输出,不满足。。额。。当然也要输出。
小弱这里先三次遍历做判断,然后再遍历三遍输出答案。(突然觉得好麻烦),对了还要注意一下特殊测试,n= 2&& abs(a[1] - a[2]) == 4,这个是不满足要求的。复杂度:
时间复杂度:O(n)
空间复杂度:O(n)代码
/* ***********************************************
Author :Ilovezilian
Created Time :2015/8/7 14:40:19
File Name :1001_2.cpp
************************************************ */
#include <bits/stdc++.h>
#define fi(i,n1,n) for(int i = n1; i <= n; i ++)
#define fd(i,n1,n2) for(int i = n1; i >= n2; i --)
#define INF 0x7fffffff
#define ll long long
using namespace std;
const int N = 100100, mod = 1e9+7;
int a[N], b[N], av, ans, n;
bool solve1()
{
ans = 0;
fi(i,2,n)
{
int r = i + 1;
if(r > n) r = 1;
if(a[i] > 0) a[i] --, a[r] ++, ans ++;
else if(a[i] < 0) a[i] ++, a[r] --, ans ++;
}
///fi(i,1,n) printf("a[%d] = %d\n" ,i, a[i]);
fi(i,1,n) if(a[i]) return 0;
return 1;
}
void prt()
{
fi(i,2,n)
{
int r = i + 1;
if(r > n) r = 1;
if(b[i] > 0) b[i] --, b[r] ++, printf("%d %d\n", i, r);
else if(b[i] < 0) b[i] ++, b[r] --, printf("%d %d\n", r, i);
}
}
bool judge()
{
if(solve1())
{
printf("YES\n%d\n", ans);
prt();
//printf("3test\n");
return 1;
}
memcpy(a, b, sizeof(b[0]) * (n + 1));
if(a[1] >= 0 && a[1] <= 2)
{
a[1] --, a[2] ++;
if(solve1())
{
b[1] --, b[2] ++;
printf("YES\n%d\n", ans+1);
printf("1 2\n");
prt();
//printf("1test\n");
return 1;
}
}
memcpy(a, b, sizeof(b[0]) * (n + 1));
if(a[1] <= 0 && a[1] >= -2)
{
a[1] ++, a[2] --;
if(solve1())
{
b[1] ++, b[2] --;
printf("YES\n%d\n", ans+1);
printf("2 1\n");
prt();
//printf("2test\n");
return 1;
}
}
//printf("4test\n");
return 0;
}
void solve()
{
ll sum = 0;
scanf("%d", &n);
fi(i,1,n)
{
scanf("%d", a + i);
sum += a[i];
}
/*个人觉得这个可以不用,是一种剪枝啦!
bool ok = 1;
fi(i,1,n) if(a[i] != a[i-1]) { ok = 0; break;}
if(ok)
{
printf("YES\n0\n");
return;
}
*/
if(sum % n || (n == 2 && abs(a[2] - a[1]) == 4))//特判前面那个其实也可以不用的。
{
printf("NO\n");
return;
}
av = sum / n;
fi(i,1,n) b[i] = a[i] = a[i] - av;
if(!judge())
{
printf("NO\n");
return;
}
}
int main()
{
int cas;
scanf("%d", &cas);
while(cas --) solve();
return 0;
}