B. Restore Modulo
题面:
中文题意:
给定一个数组,如果里面的数满足ai=(ai-1+c) mod m,且a1=s mod m。如果给定数组能找到这样的c和m就输出,如果不存在则输出-1,输出0代表m可以无限大。
思路:
统计不同的差值个数,想到用set
维护。
- 若不同差值超过2个则输出-1。
- 若不同差值只有1个或者n=1输出0。
- 其他情况,c为set中大于零的部分,m为set中两个值的绝对值相加。
代码:
//题意 求经过特定3点的最小生成树
//思路:枚举任何一点作为支撑点 ,特定的3点要相连必须经过共同的一点 ,求这三点到所枚举点的最短路和,取最小值即为答案
#include<iostream>
#include<stdio.h>
#include<string.h>
#include <random>
#include<queue>
#include <map>
#include <vector>
#include <algorithm>
#include <set>
using namespace std;
typedef pair<int, int> P;
typedef long long ll;
const int maxn = 1e5 + 39;
ll a[maxn];
ll b[maxn];
set<ll> s;
bool flag;
int main() {
ios::sync_with_stdio(false);
cin.tie(0), cout.tie(0);
int T;
cin >> T;
while (T--) {
s.clear();
int n;
ll sum1 = 0;
ll sum2 = 0;
ll c;
ll m = 0;
cin >> n;
ll big = 0;
for (int i = 1; i <= n; i++) {
cin >> a[i];
big = big > a[i] ? big : a[i];
}
for (int i = 2; i <= n; i++) {
int temp = a[i] - a[i - 1];
s.insert(temp);
}
if (s.size() == 1 || s.size() == 0) {
//注意s.size()==0,否则1 1 1会输出-1,实际上应该是0.
cout << 0 << endl;
continue;
} else if (s.size() > 2) {
cout << -1 << endl;
continue;
}
for (auto it : s) {
if (it >= 0) {
c = it;
sum1++;
} else
sum2++;
m += it >= 0 ? it : -it;
}
if (m >= big && sum1 == 1 && sum2 == 1)
cout << m << " " << c << endl;
else
cout << -1 << endl;
}
}
//OK