[NICA #2] 回溯的雨
题目描述
小 Ran 有两个正整数 x , y x,y x,y,她还有两个长度为 n n n 的正整数序列 a a a 与 c c c 满足 a i x + y = c i a_ix+y=c_i aix+y=ci ( 1 ≤ i ≤ n 1\le i\le n 1≤i≤n)。
很可惜的是,她忘记了所有 a i a_i ai 的值与 y y y 的值。她给你了 x x x 的值与序列 c c c,并希望你能还原出一组 y y y 的值与序列 a a a 使得原先的条件被满足。
当然,可能存在很多组这样的解,所以她希望你能最大化 y y y 的值并输出它(注意到你并不需要输出 a a a 序列)。如果不能还原出这样一组数,请输出 − 1 -1 −1 来报告无解。
输入格式
第一行两个正整数 n n n 与 x x x,含义同题目描述。
第二行 n n n 个正整数,描述数组 c c c,其中第 i i i 个数即为 c i c_i ci。
输出格式
输出一个正整数,代表你还原出的最大的 y y y 的值。如果不能还原出这样一组正整数,则输出 − 1 -1 −1。
样例 #1
样例输入 #1
5 2
4 6 8 10 12
样例输出 #1
2
样例 #2
样例输入 #2
5 2
3 6 9 12 15
样例输出 #2
-1
提示
数据保证, 1 ≤ n ≤ 1 0 5 1 \leq n \leq 10^5 1≤n≤105, 1 ≤ x ≤ 1 0 12 1 \leq x \leq 10^{12} 1≤x≤1012, 1 ≤ c i ≤ 1 0 18 1 \leq c_i \leq 10^{18} 1≤ci≤1018。
原始人思路是从最大的y开始到最小正整数也就是1结束找到满足条件(对于每c中每一个数,使之减去y再对x取余为0)的y,
找到就输出,结束程序,找不到,输出-1,结束程序.[简单明了,理想化a序列,y要足够大,只要(ci-y)%x为正整数即可.]
显然基本逻辑过得去,但时间复杂度极高,而本题恰好就考察这里
先上莽夫写法:
#include<bits/stdc++.h>
using namespace std;
int pd(int s[],int a,int j,int n){
for(int i=1;i<n;i++){
if((s[i]-j)%a!=0)return 0;
}
return 1;
}
int main(){
int n;
int a;
int flag=0;
//scanf("%d%d",&n,&a);
cin>>n>>a;
int s[n];
for(int i=0;i<n;i++){
cin>>s[i];
}
for(int i=0;i<n;i++){
for(int j=1;j<s[i];j++){
if((s[i]-j)%a==0){
if(pd(s,a,j,n)){
cout<<j;
flag=1;
return 0;
}
}
}
}
if(flag==0)cout<<-1;
}
一看样例全对,一提交全部超时
改进一下:
#include<iostream>
#include<algorithm>
#include<cmath>
using namespace std;
#define int long long
int n, x, c[100001];
signed main() {
ios::sync_with_stdio(0);
cin >> n >> x;
for ( int i = 1; i <= n; ++i) cin >> c[i];
sort(c + 1, c + n + 1);
if (c[1] < x) {
cout << "-1\n";
return 0;
}
for ( int i = 2; i <= n; ++i) {
c[i] -= c[1];
if (c[i] % x != 0) {
cout << "-1\n";
return 0;
}
}
cout << c[1] - x << endl;
return 0;
}
忽略y,以排序后的c1为基础(对应的a1为1),往后的ci减去c1只要都能被x整除,c1-x即为最大y,此法优化了时间,减少了循环
如有疑问或更好的解法,欢迎提出哦!