【CF618F】Double Knapsack（抽屉原理）（前缀和）

传送门

---

题解：

注意到题目给了一个很诡异的限制，每个元素的大小不超过集合大小。

冷静分析。。。感觉一定有解。

把总和较大的集合设为$S_1$，另一个设为$S_2$，计算两个序列的前缀和（从$0$开始）记为 { s 1 } , { s 2 } \{s_1\},\{s_2\} {s1​},{s2​}，考虑$s_2$中的每个元素，我们在$s_1$中找到它的lower_bound（即大于等于它的最小数），作差得到的结果显然在$[0,n-1]$中。

注意到一共有$n$种不同的结果，但是我们是从$0$开始一共计算了$n+1$次。也就意味着差的结果一定有重复。

找到上一次重复的地方，两个前缀和之差中间的部分就相等。

---

代码：

#include<bits/stdc++.h>
#define ll long long
#define re register
#define cs const

namespace IO{
	inline char gc(){
		static cs int Rlen=1<<22|1;
		static char buf[Rlen],*p1,*p2;
		return (p1==p2)&&(p2=(p1=buf)+fread(buf,1,Rlen,stdin),p1==p2)?EOF:*p1++;
	}
	template<typename T>T get(){
		char c;T num;
		while(!isdigit(c=gc()));num=c^48;
		while(isdigit(c=gc()))num=(num+(num<<2)<<1)+(c^48);
		return num;
	}inline int gi(){return get<int>();}
}
using namespace IO;

using std::cerr;
using std::cout;

cs int N=1e6+7;
bool vis[N],flag;
int p1[N],p2[N],n;
ll s1[N],s2[N];
int b1,b2,e1,e2;

signed main(){
#ifdef zxyoi
	freopen("double_knapsack.in","r",stdin);
#endif
	n=gi();
	for(int re i=1;i<=n;++i)s1[i]=s1[i-1]+gi();
	for(int re i=1;i<=n;++i)s2[i]=s2[i-1]+gi();
	if(s1[n]<s2[n])std::swap(s1,s2),flag=true;
	for(int re i=0,j=0;i<=n;++i){
		while(s1[j]<s2[i])++j;
		int t=s1[j]-s2[i];
		if(vis[t]){
			int b1=p1[t],b2=p2[t],e1=i,e2=j;
			if(!flag)std::swap(b1,b2),std::swap(e1,e2);
			cout<<e1-b1<<"\n";
			for(int re i=b1+1;i<=e1;++i)cout<<i<<" ";
			cout<<"\n"<<e2-b2<<"\n";
			for(int re i=b2+1;i<=e2;++i)cout<<i<<" ";
			return 0;
		}
		vis[t]=true;
		p1[t]=i;p2[t]=j;
	}cerr<<"error?\n";
	return 0;
}