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

 1 #include<iostream>
 2 #include<algorithm>
 3 #include<map>
 4 #include<vector>
 5 using namespace std;
 6 
 7 int board[100005];
 8 vector<int> va;
 9 map<int,int> m;
10 long long ans;
11 //贪心策略
12 //每个都向自己最近的左边的棋子移动
13 int main(){
14     int n; cin>>n;
15     for(int i=1;i<=n;i++) cin>>board[i];
16     for(int i=1;i<=n;i++){
17         int x; cin>>x;
18         if(x){//这个位置上有棋子
19             va.push_back(i);
20             m[i]=x;
21         }
22     }
23     
24     for(int i=1;i<=n;i++){
25         while(1){
26             vector<int>::iterator index = upper_bound(va.begin(),va.end(),i);/* index指向第一个大于i的元素 */
27             //如果没有比i还大的，返回begin()
28             if(index==va.begin()) break;//目标盘左边没有棋子了 
29             index--; 
30             if( m[ *index ]>board[i] ) {
31                 ans+=(i-(*index) )*board[i];
32                 m[*index]-=board[i];
33                 board[i]=0;
34                 break;
35             }
36             if( m[ *index ]==board[i] ){
37                 ans+=(i-(*index))*board[i];
38                 va.erase(index);
39                 board[i]=0;
40                 break;
41             }
42             if( m[ *index]<board[i]){
43                 ans+=(i-(*index))*m[ *index];
44                 va.erase(index);
45                 board[i]-=m[*index];
46             }
47         }
48     }
49     
50     for(int i=1;i<=n;i++){
51         if(board[i]==0) continue;
52         while(1){
53             if( m[ va[0] ]>board[i] ) {
54                 ans+=(i+va[0]-2)*board[i];
55                 m[ va[0] ]-=board[i];
56                 break;
57             }
58             if( m[ va[0] ]== board[i] ){
59                 ans+=(i+va[0]-2)*board[i];
60                 va.erase(va.begin());
61                 break;
62             }
63             if( m[ va[0] ]<board[i]){
64                 ans+=(i+va[0]-2)*m[ va[0] ];
65                 va.erase(va.begin());
66                 board[i]-=m[ va[0] ];
67             }
68         }
69     }
70     
71     cout<<ans<<endl;
72     
73     return 0;
74 }