ZJNU 2135 - 小智的宝可梦

因为成环

所以可以枚举第1只与第n只喂的次数

然后第1只和第2只的次数就固定了，以此类推，则所有宝可梦喂的次数都固定了

最后处理完检查是否全为0，不是则进行下一次枚举，是则直接输出Yes

如果所有枚举均不成立，输出No

#include<bits/stdc++.h>
using namespace std;
int main(){
    ios::sync_with_stdio();
    cin.tie();cout.tie();
    int n,i,j,a[],d[];
    bool pass;
    cin>>n;
    if(n<){
        cout<<"Yes\n";
        return ;
    }
    else if(n==){
        cin>>a[]>>a[];
        if(a[]!=a[])
            cout<<"No\n";
        else
            cout<<"Yes\n";
        return ;
    }
    for(i=;i<=n;i++)
        cin>>a[i];
    for(j=;j<=a[];j++){//枚举第1只与第n只喂的次数
        for(i=;i<=n;i++)
            d[i]=a[i];
        d[]-=j;
        d[n]-=j;
        for(i=;i<n;i++){
            d[i+]-=d[i];
            d[i]=;
        }
        pass=true;
        for(i=;i<=n;i++)
            if(d[i]){
                pass=false;
                break;
            }
        if(pass){
            cout<<"Yes\n";
            return ;
        }
    }
    cout<<"No\n";

    return ;
}