Codeforces Round #632 (Div. 2) B Kind Anton 【1333B】(思维题)

Kind Anton

题意

给定一个数组 a ，a 中只含有 {-1，0，1} 这三个数，可以执行以下操作：

1. 选一个$（i，j）$,$1\le i<j\le n$,可以被选多次
2. 将 $a[j]$ 变为 $a[j]+a[i]$
   然后给定一个数组 b ，问 a 能否根据以上操作变成 b 。

思路

对于 a 中的一个元素 a[x] 而言，如果它前面有 1，那么 a[x] 是可以增加的，如果前面有 -1 ，那么 a[x] 是可以减少的。
所以我们设定两个状态，遍历数组，x表示可增加，y表示可减少。
对于一个 i， 如果 $b[i]>a[i]$ 并且状态 可增加 是假的，那么a不能变成b。
同理，如果 $b[i]<a[i]$ 并且状态 可减少 是假的，那么a也不能变成b。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<string>
 
using namespace std;
 
int a[100005];
int b[100005];
 
int main()
{
    int t;
    cin >> t;
    int n;
    while(t--){
        cin >> n;
        for(int i = 1; i <= n; ++i)
            cin >> a[i];
        for(int i = 1; i <= n; ++i)
            cin >> b[i];
        int x = 0;//为1时可增加
        int y = 0;//为1时可减少
        int flag = 0;
        for(int i = 1; i <= n; ++i){
            if(a[i]!=b[i]){
                if(b[i]>a[i]&&x==0){
                    flag = 1;
                    break;
                }
                if(b[i]<a[i]&&y==0){
                    flag = 1;
                    break;
                }
            }
            if(a[i]==1)
                x = 1;
            if(a[i]==-1)
                y = 1;
        }
        if(flag)
            printf("NO\n");
        else
            printf("YES\n");
    }
 
    return 0;
}

碎碎念

我觉得这题比A题简单多了…