前言
坚持每天做一道算法题,每天学一点数据结构与算法,写思路和题解来加深自己的印象。
题目描述
有 2n (n≤7) 个国家参加世界杯决赛圈且进入淘汰赛环节。已经知道各个国家的能力值,且都不相等。能力值高的国家和能力值低的国家踢比赛时高者获胜。1 号国家和 2 号国家踢一场比赛,胜者晋级。3 号国家和 4 号国家也踢一场,胜者晋级……晋级后的国家用相同的方法继续完成赛程,直到决出冠军。给出各个国家的能力值,请问亚军是哪个国家?
输入格式
第一行一个整数 n ,表示一共 2n 个国家参赛。
第二行 2n 个整数,第 i 个整数表示编号为 i 的国家的能力值(1 ≤ i ≤ 2n )。
数据保证不存在平局。
输出格式
仅一个整数,表示亚军国家的编号。
输入输出样例
输入1
3
4 2 3 1 10 5 9 7
输出1
1
解题思路
快速幂+结构体数组排序(还可以二分+递归、STL的队列)
AC代码
#include <iostream>
#include <algorithm> // 调用sort函数
using namespace std;
struct node {
int id; // 国家的编号
int strength; // 国家的能力值
};
int n, all;
node data[10000000]; // 结构体数组
const int mod = 1e9 + 7;
// 快速幂
int quickpow(int a, int n) {
int sum = 1;
while(n) {
if(n & 1) sum = sum * a % mod;
a = a * a % mod;
n >>= 1;
}
return sum;
}
// sort的自定义比较函数
bool cmp(node a, node b) {
return a.strength > b.strength; // 降序输出
}
int main() {
cin >> n;
all = quickpow(2, n);
for(int i = 1; i <= all; i++) {
data[i].id = i;
cin >> data[i].strength;
}
sort(data + 1, data + all / 2 + 1, cmp);
sort(data + all / 2 + 1, data + all + 1, cmp);
if(data[1].strength > data[all / 2 + 1].strength) cout << data[all / 2 + 1].id;
else cout << data[1].id;
return 0;
}
总结
我看大佬的题解,好像直接用pow函数也可以。那我写个快速幂就当作是复习模板好了。还有好多好多解法,不过我一开始想到的就是结构体数组排序+快速幂了。算法之路很漫长~~