【2020】牛客模考（一模）编程题1-数组变换

目录

[编程题]数组变换

分析：

法一：普通方法

法二：利用bitset，截断字符串

---

---

1/8

[编程题]数组变换

时间限制：C/C++ 1秒，其他语言2秒

空间限制：C/C++ 32M，其他语言64M

牛牛有一个数组，里面的数可能不相等，现在他想把数组变为：所有的数都相等。问是否可行。
牛牛可以进行的操作是：将数组中的任意一个数改为这个数的两倍。
这个操作的使用次数不限，也可以不使用，并且可以对同一个位置使用多次。
 

输入描述:

输入一个正整数N (N <= 50)
接下来一行输入N个正整数，每个数均小于等于1e9.

输出描述:

假如经过若干次操作可以使得N个数都相等，那么输出"YES", 否则输出"NO"

输入例子1:

2
1 2

输出例子1:

YES

分析：

法一：普通方法

找出数组a中最大的数，然后再遍历数组，对于每个数a[i]，先用MAX对a[i]求余，余数等于0，并且商是2的幂，若每一个数组元素都满足这个条件，则通过x2最终可以得到所有数相等的数组。

#include <iostream>
#include <math.h>
#include <bitset>
using namespace std;
int main(){
    int n;
    cin>>n;
    int b[n];
    int MAX=0xffffffff;
    int index=0;
    for(int i=0;i<n;i++){
        cin>>b[i];
     if(b[i]>MAX){
            MAX=b[i];
            index=i;
        }
    }
    for(int i=0;i<n;i++){
        //最大值能除尽a[i]
        if(i!=index&&MAX%b[i]==0){
            //并且，商是2的幂
            double s=(double)MAX/(double)b[i];
            double t=log(s)/log(2);
            //判断t是否是整数，由于double有精度缺失
            if(abs(round(t) - t) < 0.000000000000001) continue;
            else {
                cout<<"NO"<<endl;
                return 0;
            }
        }
    }
    cout<<"YES"<<endl;
    return 0;
}

这个代码只通过了5个，还存在一些bug

法二：利用bitset，截断字符串

由于题目的特殊性，乘以2就相当与左移一位，末位补0/1。现在题目要求任意次乘2，即左移任意位补任意个0，左边的0也是任意个，只需要从第一个1到最后一个1中间截断的字符串是一样的，那么完全可以通过左移末位补0，得到相同的数。

算法步骤：将输入n个十进制数字转换成二进制字符串，然后截取第一个'1'到最后一个'1'中间的字符串，判断n个数字的截断字符串是否相等。

其中将十进制数转成二进制字符串时利用bitset，bitset是STL里定义的模板，bitset是一个布尔数组，每个布尔值只占用1bit，所以利用它来表示二进制字符串是很方便的。

ps：这里可能需要稍微注意一下bitset变量的输入，和set，vector一样，这个不能直接使用cin>>进行输入

可以如下声明一个该类型变量：

bitset<N>varm (M)

其中varm为变量名。

N表示该类型在内存中占的位数，是二进制。

M表示变量varm的初始值。

#include <iostream>
#include <math.h>
#include <bitset>
using namespace std;
int main(){
    int n,b;
    cin>>n;
    string s,pre;
    for(int i=0;i<n;i++){
        cin>>b;
        bitset<32> a(b);    //括号里放输入的变量，bitset要注意，不能直接输入bitset变量cin>>a,而应该把变量放在括号里做初始化
        
        s=a.to_string();    //转换成字符串
        int first=s.find('1');
        int last=s.rfind('1');
        //为什么输入2,4，得到的字符串都是0000000000000000000？？？？？a的输入方式不对？（已解决）
        //截断字符串，从第一个1到最后一个1
        s=s.substr(first,last-first+1);    
        //cout<<first<<last<<' '<<s<<endl;
        //若前一个和后一个截断后字符串相等
        if(i!=0&&s==pre){
            continue;
        }
        //但凡有一个不等，都说明不能通过x2，即左移一位得到相同的数
        else if(i!=0&&s!=pre){
            cout<<"NO"<<endl;
            return 0;
        }
        else{
            pre=s;
        }
    }
    cout<<"YES"<<endl;
    return 0;
}

代码通过