1056. Mice and Rice (25)

1056. Mice and Rice (25)

/*
题目意思是：
n个人，每人手中拿着一个数字，然后按照k人一组进行比赛，谁的数大谁就赢，最后输出他们的排名

输入说明：
第一行是n,k，分别代表参赛人数和一个组的最大人数
第二行是n个数据，比如这里，就是从0到10号参赛选手的手中的数字
第三行是n个参赛选手的顺序，分组就是依据此来划分

详细说明：

11 3 //11个参赛选手，3人一组

以下数据中，第一行代表参赛选手的序号，第二行代表参赛选手手中的数字

选手序号：00 01 02 | 03 04 05 | 06 07 08 | 09 10
手中数字：25 18 00 | 46 37 03 | 19 22 57 | 56 10

第一局：
比赛次序：06 00 08 | 07 10 05 | 09 01 04 | 02 03   rank->06 00 10 05 01 04 02 :5
对应数字：19 25 57 | 22 10 03 | 56 18 37 | 00 46

第二局：
08 07 09 | 03   rank->07 09 :3
57 22 56 | 46

第三局：
08 03   rank->03:2
57 46

第四局：
08   rank->08:1
57

结果：
00 01 02 03 04 05 06 07 08 09 10
 5  5  5  2  5  5  5  3  1  3  5 

思路：
对于每一局比赛，我知道可以分成多少个组，那么，我同样就会知道会有多少个人晋级（每组晋级一人）
那么，剩下被淘汰的人就是同一排名了，并且排名也知道（等于晋级人数加一，剩下一人时为1）
比如，第一局，分成四个小组，四人晋级，剩下被淘汰的人排名为5.
好了，开始我们的代码吧:)
*/
#include <bits/stdc++.h>
using namespace std;
vector<int> nums,ranks;
vector<int> MiceAndRice(vector<int> &v,int k)
{
    int n=v.size();//本局参赛人数
    int groupnum=(n%k==0)?n/k:n/k+1;//一个多少个组
    vector<int> nextv;//晋级选手
    for(int i=1;i<=groupnum;++i)//每个小组进行比赛
    {
        auto itb=v.begin()+(i-1)*k,ite=min(v.end(),itb+k);//当前组成员范围
        auto it=itb,maxit=itb;
        while(it!=ite)
        {
            if(nums[*maxit]<nums[*it])
                maxit=it;
            ++it;
        }
        nextv.push_back(*maxit);//有最大数字的选手晋级
        while(itb!=ite) ranks[*itb++]=groupnum+1;//排名
    }
    return nextv;
}
int main(int argc, char const *argv[])
{
    int n,k;
    cin>>n>>k;
    nums.resize(n);
    ranks.resize(n);
    vector<int> v(n);
    for(int i=0;i<n;++i)
        cin>>nums[i];
    for(int i=0;i<n;++i)
        cin>>v[i];
    while(v.size()>1) v=MiceAndRice(v,k);
    ranks[v[0]]=1;
    for(int i=0;i<n;++i)
        i==n-1?cout<<ranks[i]:cout<<ranks[i]<<" ";
    return 0;
}