NYOJ109 数列转换 【守恒法】

数列转换

时间限制： 3000 ms  |  内存限制： 65535 KB

难度： 3

描述

有一个数列a₁,a₂,a₃...a_n,每次可以从中任意选三个相邻的数a_i-1 ,a_i , a_{i+1 ,}进行如下操作（此操作称为“对a_i进行操作”）

(a_i-1,a_i,a_i+1)->(a_i-1+a_i,-a_i,a_i+a_i+1)

给定初始和目标序列，是否能通过以上操作，将初始序列转换成为目标序列？例如，初始序列(1 6 9 4 2 0)目标序列（7 -6 19 2 -6 6)可经过如下操作：

 (1 6 9 4 2 0)->( 1 6 13 -4 6 0)->(1 6 13 2 -6 6)->(7 -6 19 2 -6 6)

请你判断给定的初始状态和目标状态，输出Yes（能够转换）或No（不能转换）

输入

第一行是一个正整数N,表示测试数据的组数。(N<=100)
每组测试数据的第一行是一个整数M(3<=M<=1000),表示该组测试数据的起始状态与结束状态都有M个数。
每组测试数据的第二行是M个整数Ai(-1000<=Ai<=1000)，表示起始状态。
每组测试数据的第三行是M个整数Bi(-1000<=Bi<=1000),表示终止状态。

输出

如果能够转换，输出Yes
如果不能转换，输出No

样例输入

2
3
1 2 3
1 3 2
6
1 6 9 4 2 0
7 -6 19 2 -6 6

样例输出

No
Yes

来源

《世界大学生程序设计竞赛高级教程·第一册》

上传者

张云聪

题解：如果用暴力模拟出所有结果然后再比较的话很容易超时，而且编程复杂度高，但是在网上看到了一个很神奇的守恒法，不是在细节上关注这些变化，而是从整体上看到某些不变的东西，用S[i]表示A[0...i]的和，当对A[i]进行变化的时候，直接结果是S[i]和S[i-1]交换了位置，所以为题就转换成了求S数组，排序，然后比较两个S数组是否相等。

#include <stdio.h>
#include <string.h>
#include <algorithm>

#define maxn 1010

int A[maxn], B[maxn];

int main() {
    int t, n, i;
    scanf("%d", &t);
    while(t--) {
        scanf("%d", &n);
        for(i = 0; i < n; ++i) {
            scanf("%d", &A[i]);
            if(i) A[i] += A[i-1];
        }
        for(i = 0; i < n; ++i) {
            scanf("%d", &B[i]);
            if(i) B[i] += B[i-1];
        }
        std::sort(A, A + n);
        std::sort(B, B + n);
        if(std::equal(A, A + n, B))
            printf("Yes\n");
        else printf("No\n");
    }
    return 0;
}