Gipsy小姐姐有一个数列,数列总和共有n个数,每个数的值为Ci,但是Gipsy并不喜欢它,她每次可以选择一个数i(2<=i<=n-1),并将Ci的值变为C[i]`=C[i+1]+C[i-1]-C[i],Gipsy可以这样caoz操作无数次,直到对于任意的Ci,其值变为Gipsy希望的值Ti
输入格式:
第一行输入一个n,n的范围在2到100000之间
第二行输入n个数,表示数列里这n个数的初始值C1,C2...Cn
第三行输入n个数,表示Gispy希望数列里这n个数变成的值T1,T2...Tn
每个数的范围在0到1000000000
输出格式:
如果存在可能(数列可能为空),使得这个数列变成Gispy期望的样子,输出Yes,否则No
输入样例:
Input
4
7 2 4 12
7 15 10 12
Output
Yes
#include <iostream>
#include <vector>
using namespace std;
vector<int> C, T, vis;
struct Temp { //便于回溯复原
int pre, post, c;
};
bool Gipsy() {
bool result = false;
if (C == T)
return true;
for (int i = 1; i < C.size()-1; i++) {
if (vis[i] == 0&& C[i - 1] + C[i + 1] - C[i]<=T[i]) {
vis[i] = 1; //标记访问
Temp t = { vis[i - 1],vis[i + 1],C[i] };
vis[i - 1] = vis[i + 1] = 0; //相邻数可以继续访问
C[i] = C[i - 1] + C[i + 1] - C[i];
result=result||Gipsy();
C[i] = t.c; //回溯
vis[i] = 0;
vis[i - 1] = t.pre;
vis[i + 1] = t.post;
}
}
return result;
}
int main() {
int n;
cin >> n;
C.resize(n); T.resize(n);
vis.resize(n, 0);
for (int i = 0; i < n; ++i)
cin >> C[i];
for (int i = 0; i < n; ++i)
cin >> T[i];
cout << (Gipsy() ? "Yes" : "No") << endl;
return 0;
}
/*
4
7 2 4 12
7 15 10 12
*/