Codeforces Round #796 (Div. 2) F. Sanae and Giant Robot

最新推荐文章于 2022-07-27 18:27:52 发布

yl-9

最新推荐文章于 2022-07-27 18:27:52 发布

阅读量263

点赞数 1

分类专栏： Codeforces 文章标签： c++

本文链接：https://blog.csdn.net/qq_40329819/article/details/125135266

版权

Codeforces 专栏收录该内容

7 篇文章 0 订阅

订阅专栏

题目链接

题意

给两个长度为 $n$ 的数组 $a$ 和 $b$ ，以及 $m$ 个区间 $[l i, r i]$ 。每次你可以从这 $m$ 个区间中选择一个满足 $\sum_{i=l}^ra_i=\sum_{i=l}^rb_i$ 的区间 $(l, r)$ ，然后对于 $l \leq i \leq r$ 令 $a_i=b_i$ 。问通过上述操作是否可以将 $a$ 变为 $b$ 。

题解

令 $s_i=s_{i-1}+a_i-b_i$ ，则区间 $[l, r]$ 可以交换当且仅当 $s_{l−1}=s_r$ ，且交换过后对于 $s_i=s_{l−1} (l≤i≤r)$ 。目标就是让 $s_i=0$ 。

因为最后的目标是让所有位置都为零，所以置非零的数没有意义，所以可以将操作看成：对于区间 $[l, r]$ ，如果 $s_{l−1}=s_r=0$ 就将区间内的 $s_i$ 都置 $0$ ，最后看还有没有非零的 $s_i$ 就可以了。

可以对于每个位置，维护一个以这个位置为端点的区间的集合，然后枚举所有为 $s_i=0$ 的位置，再枚举以它为端点的区间，看是否可以将这个区间是否可以操作。

代码

#include<bits/stdc++.h>
#define LL long long
#define pb push_back
using namespace std;
const int N=2e5+9;
int T,n,m;
LL a[N],b[N],s[N];
set<int>nz;
queue<int>q;
vector<int>e[N];
void solve()
{
	cin>>n>>m;
	nz.clear();
	while(!q.empty()) q.pop();
	for(int i=1;i<=n;i++)
		cin>>a[i];
	for(int i=1;i<=n;i++)
	{
		cin>>b[i];
		s[i]=s[i-1]+a[i]-b[i];
		nz.insert(i);
		e[i].clear();
	}
	e[0].clear();
	for(int i=1,x,y;i<=m;i++)
	{
		cin>>x>>y;
		x--;
		e[x].pb(y);
		e[y].pb(x);
	}
	nz.insert(0);
	for(int i=0;i<=n;i++)
		if(!s[i])
		{
			q.push(i);
			nz.erase(i);
		}
	while(!q.empty())
	{
		int x=q.front(); q.pop();
		for(int y:e[x])
		{
			if(s[y]) continue;
			int l=min(x,y),r=max(x,y);
			auto itl=nz.lower_bound(l),itr=nz.upper_bound(r);
			for(auto i=itl;i!=itr;i++)
				s[*i]=0,q.push(*i);
			nz.erase(itl,itr);
		}
	}
	cout<<(nz.empty()?"YES\n":"NO\n");
}
int main()
{
	ios_base::sync_with_stdio(false); cin.tie(0); cout.tie(0);
	cin>>T;
	while(T--) solve();
	return 0;
}