洛谷刷题笔记Q2 欢乐的跳 使用set

题目链接

欢乐的跳

题目描述

一个 $n$ 个元素的整数数组，如果数组两个连续元素之间差的绝对值包括了 $[1,n-1]$ 之间的所有整数，则称之符合“欢乐的跳”，如数组 { 1 , 4 , 2 , 3 } \{1,4,2,3\} {1,4,2,3} 符合“欢乐的跳”，因为差的绝对值分别为：$3,2,1$。

给定一个数组，你的任务是判断该数组是否符合“欢乐的跳”。

输入格式

每组测试数据第一行以一个整数 $n(1\le n\le 1000)$ 开始，接下来 $n$ 个空格隔开的在 $[-10^8,10^8]$ 之间的整数。

输出格式

对于每组测试数据，输出一行若该数组符合“欢乐的跳”则输出 Jolly，否则输出 Not jolly。

样例 #1

样例输入 #1

4 1 4 2 3

样例输出 #1

Jolly

样例 #2

样例输入 #2

5 1 4 2 -1 6

样例输出 #2

Not jolly

提示

$1\le n\le 1000$

思考

考虑将相邻差值去重地存进一个变量b，再将[1,n-1]之间的数与b中元素比对。因此考虑使用集合set，这个类型存放差值，因为set要求元素不重复。

##代码

#include<iostream>
#include<vector>
#include<algorithm>
#include<set>
using namespace std;

int main(){
    //freopen("a.in","r",stdin);
    int n;
    cin>>n;
    vector<int> a;
    int tmp;
    for(int i=0;i<n;i++){
        cin>>tmp;
        a.push_back(tmp);
    }

    set<int> b;
    for(int i=0;i<n-1;i++){
        b.insert(abs(a[i+1]-a[i]));
    }

    bool flag=true;
    for(int i=1;i<n;i++){
        if(b.count(i)!=true){
            flag=false;
            break;
        }
    }
    if(flag==true){
        cout<<"Jolly"<<endl;
    }
    else 
        cout<<"Not jolly"<<endl;
    return 0;
}