Codeforces 733C Epidemic in Monstropolis 构造+思维

点击打开链接

//b的k个中的第一个若从a[x']开始合并 因为b[1]为第一个,所以a[1~x']必须全部合并完,

//如果sum[1~x']<b[1]则只能继续向右合并&&因为只能和相邻的合并,所以最后b[1]=sum[1~x] 为a的一段前缀和 
//同理后面的b[j]为a[i]的一段连续和 

b[j]满足前缀和时 构造解即可:从一块的最大值开始向相邻的合并,合并时移动编号

#include <iostream>
#include <algorithm>
#include <list>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <queue>
#include <vector>
using namespace std;
typedef long long ll;
const ll mod=1e9+7;
const int N= 1e6+20;
ll a[N],b[N];
vector<ll> v[N];//分块 
vector<pair<ll,ll> > ans; 
//因为只能合并相邻的 剩下的k个中的第一个肯定为1~x的之和 
//第二个为x+1~y之和,同理后面的b[j]为a[i]的一段连续和 
int main()
{
	int n,k;
	cin>>n;
	ll x=0,y=0;
	for(int i=1;i<=n;i++)
	{
		cin>>a[i];
		x+=a[i];
	}
	cin>>k;
	int l=1;
	bool flag=true;
	for(int j=1;j<=k;j++)
	{
		cin>>b[j];
		y+=b[j];
	}
	if(x!=y)//数量不同肯定无解 
	{
		cout<<"NO"<<endl;
		return 0;
	}
	ll s=0,cnt=1;
	for(int i=1;i<=n;i++)
	{
		v[cnt].push_back(a[i]);
		s+=a[i];	
		if(s>b[cnt])
		{
			flag=false;
			break;
		}
		if(s==b[cnt])
		{
			s=0;
			cnt++;
		}
	}

	if(!flag)
	{
		cout<<"NO"<<endl;
		return 0;
	}
	else
	{
		for(int i=1;i<=k;i++)//构造解 ://最大值如果和相邻合并后,大于该块所有值,扫一遍即可 
		{
			if(v[i].size()==1)
			continue;
			
			
			bool flag=false;
			ll t=v[i][0];
			ll mx=t;
			for(int j=1;j<v[i].size();j++)
			{
				if(t!=v[i][j])//该块全部相同无法合并 
				{
					flag=true;//
				}
				mx=max(mx,v[i][j]);//找到最值开始合并 
			}
			if(!flag)
			{
				cout<<"NO"<<endl;
				return 0;	
			}
			
			int pos,lr=0;//标记最大值旁边是否有比他小的数
			for(int j=0;j<v[i].size()-1;j++)//判定最值能否和相邻的合并 
			{
				if(v[i][j]==mx&&v[i][j+1]<mx)
				{
					lr=1;
					pos=j;
					break;
				}
			}
			if(!lr)
			{
				for(int j=v[i].size()-1;j>0;j--)
				{
					if(v[i][j]==mx&&v[i][j-1]<mx)
					{
						lr=-1;
						pos=j;
						break;
					}
				}	
			}
			if(lr==1)//先右边
			{
			 
				for(int j=pos+1;j<v[i].size();j++)
				{
					ans.push_back(make_pair(pos+i,1));//向右移动 
				}
				for(int j=pos-1;j>=0;j--)
				{
					ans.push_back(make_pair(j+i+1,-1));//
				}
			}
			else
			{
				for(int j=pos-1;j>=0;j--)
				{
					ans.push_back(make_pair(j+i+1,-1));//
				}
				for(int j=pos+1;j<v[i].size();j++)
				{
					ans.push_back(make_pair(i,1));//向右移动 
				}	
			} 
		 
		}
		cout<<"YES"<<endl;
		for(int i=0;i<ans.size();i++)
		{
			cout<<ans[i].first<<" ";
			if(ans[i].second==-1)
			cout<<"L"<<endl;
			else
			cout<<"R"<<endl; 
		}
	}
	return 0;
}