2018美团CodeM编程大赛 Round A Problem 2 下棋 【贪心】

应该一眼看出来是贪心题，然后想最优解是什么。正确的贪心策略是【原棋盘上每个位置的棋子】都往最近的左边【目标棋盘上棋子】移动，如果左边没有棋子了那就闲置最后处理，如果目标棋盘在该位置上也有棋子，那就算距离为0（最近）。最后处理的话，棋盘上的局面应该是所有的棋子都得往右移，这样的话怎么移都无所谓了，朴素的把放过去。

模拟的时候难度在于通过upper bound找到离该位置最近左的位置，所以复杂度是O(NlogN)；如果朴素的遍历数组找最近左需要N复杂度，整体n^2就过不了了。

 对于upper_bound来说，返回的是被查序列中第一个大于查找值的指针，也就是返回指向被查值>查找值的最小指针，lower_bound则是返回的是被查序列中第一个大于等于查找值的指针，也就是返回指向被查值>=查找值的最小指针。 【引用自 https://blog.csdn.net/u011008379/article/details/50725670 】

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

int board[100005];
vector<int> va;
map<int,int> m;
long long ans;
//贪心策略
//每个都向自己最近的左边的棋子移动
int main(){
    int n; cin>>n;
    for(int i=1;i<=n;i++) cin>>board[i];
    for(int i=1;i<=n;i++){
        int x; cin>>x;
        if(x){//这个位置上有棋子
            va.push_back(i);
            m[i]=x;
        }
    }
    
    for(int i=1;i<=n;i++){
        while(1){
            vector<int>::iterator index = upper_bound(va.begin(),va.end(),i);/* index指向第一个大于i的元素 */
            //如果没有比i还大的，返回begin()
            if(index==va.begin()) break;//目标盘左边没有棋子了 
            index--; 
            if( m[ *index ]>board[i] ) {
                ans+=(i-(*index) )*board[i];
                m[*index]-=board[i];
                board[i]=0;
                break;
            }
            if( m[ *index ]==board[i] ){
                ans+=(i-(*index))*board[i];
                va.erase(index);
                board[i]=0;
                break;
            }
            if( m[ *index]<board[i]){
                ans+=(i-(*index))*m[ *index];
                va.erase(index);
                board[i]-=m[*index];
            }
        }
    }
    
    for(int i=1;i<=n;i++){
        if(board[i]==0) continue;
        while(1){
            if( m[ va[0] ]>board[i] ) {
                ans+=(i+va[0]-2)*board[i];
                m[ va[0] ]-=board[i];
                break;
            }
            if( m[ va[0] ]== board[i] ){
                ans+=(i+va[0]-2)*board[i];
                va.erase(va.begin());
                break;
            }
            if( m[ va[0] ]<board[i]){
                ans+=(i+va[0]-2)*m[ va[0] ];
                va.erase(va.begin());
                board[i]-=m[ va[0] ];
            }
        }
    }
    
    cout<<ans<<endl;
    
    return 0;
}

没有提交在美团的oj上，但自己编了几个数据都过了。