UVALive - 4977 Enter The Dragon 模拟|贪心

题目链接：点击打开链接

题目思路：用贪心的思路模拟一下，具体注释在代码中。

这题是队友写的@BinacsLee,抱紧队友大腿！在队友代码的基础上加上了注释，方便大家理解

AC代码：

/*
2017年8月22日10:21:30
UVAlive 4977
代码来自队友@BinacsLee
AC 
*/
#include <iostream>
#include <map>
#include <set>
#include <string>
#include <cstring>
#include <cstdio>
#include <algorithm>
#include <cmath>
#include <queue>
using namespace std;
const int maxn=1e6+10;
int a[maxn];//存放输入数据 
int pre[maxn];//pre[i]代表上次i水库下雨的是第几天 
struct node{
	int th,v; 
};
vector<node> ans;//存放答案 
set<int> s;//s[i]代表第i个0在第几天出现 
set<int>::iterator it;

bool cmp(node a,node b){
	return a.th<b.th;//按照出现日期排序 
}

int main(){
	int t;
	scanf("%d",&t);
	while(t--){
		int n,m;
		scanf("%d%d",&n,&m);
		/*初始化*/
		s.clear();
		ans.clear();
		memset(pre,0,sizeof(pre));
		memset(a,0,sizeof(a));
		/*输入数据*/ 
		for(int i=1;i<=m;i++){
			scanf("%d",&a[i]); 
		} 
		/*处理数据*/ 
		bool f=true; 
		for(int i=1;i<=m;i++){
			if(!f) break;
			/*第一天是雨天直接输出NO*/ 
			if(a[1]!=0){
				f=false;
			}
			/*出现晴天存进s[i]*/
			else if(a[i]==0){
				s.insert(i);
			}
			/*出现雨天，处理*/
			else{
				//从前往后找，找到第一个满足条件的晴天 
				it=s.upper_bound(pre[a[i]]);
				//找不到，输出NO 
				if(it==s.end()){
					f=false;
					continue;
				} 
				//找到了，将这点存进ans数组 
				node res;
				res.th=*it;
				res.v=a[i];
				ans.push_back(res);
				//然后把这一个晴天从s集合中去掉 
				s.erase(it);
				//这是关键点，把pre[a[i]] 置为 a[i]上一次出现的日期 
				pre[a[i]]=i; 
			}
		}
		if(!f) printf("NO\n");
		else{
			printf("YES\n");
			//如果还有没处理的晴天，存进ans 
			for(it=s.begin();it!=s.end();it++){
				node res;
				res.th=*it;
				res.v=0;
				ans.push_back(res);
			} 
			//对ans数组排序 
			sort(ans.begin(),ans.end(),cmp);
			for(int i=0;i<ans.size();i++){
				if(i!=ans.size()-1) printf("%d ",ans[i].v);
				else printf("%d\n",ans[i].v);
			}
		}
	}
	return 0;
}