题目链接(hard版):F2. Same Sum Blocks (Hard)
题目大意:给你一个序列,让你再其中选出若干个子区间(l1,r1),(l2,r2),…,(lk,rk),且子区间没有交集,每个子区间的权值和相等。问取出最多区间的方案,即最多有几个区间满足上述题意,并输出每个区间的.
注意easy版和hard版区别仅限于序列中元素个数的不同,easy版n<50,hard版n<1500.
思路:拿到这道题先想着把easy版做了再说,想到的是暴力(结果没想到竟然也是hard版的正解)。具体如下:
因为n并不大,其子序列所组合的所有值最多个,但每个值会很大,所以考虑离散化处理,想到map容器,然后我们考虑对每一个值映射一个vector来储存它的每组。然后遍历每个键值,对其所映射的vector按右端点从小到大排序,依次取出每组合法,维护答案的最大值和其vector所对应的键值或者地址(便于后续打印答案,要不然存成个数组会mle)
代码:
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>
#include <vector>
#include <map>
#include <set>
#include <queue>
typedef long long ll;
using namespace std;
#define INF 0x3f3f3f3f
const int mod=1e9+7;
const int maxn=3e7+10;
int n,a[1505];
map<int,vector<pair<int,int> > >ma;
bool cmp(pair<int,int> a,pair<int,int> b){
if(a.second==b.second)return a.first>b.first;
return a.second<b.second;
}
int main()
{
scanf("%d",&n);
int sum[1505];
sum[0]=0;
for(int i=1;i<=n;i++){
scanf("%d",&a[i]);
sum[i]=sum[i-1]+a[i];
}
for(int len=1;len<=n;len++){
for(int i=0;i+len<=n;i++){
int val=sum[i+len]-sum[i];
ma[val].push_back(make_pair(i+1,i+len));
}
}
map<int,vector<pair<int,int> > >::iterator it;
vector<pair<int,int> >as;
int ans=0;
for(it=ma.begin();it!=ma.end();it++){
vector<pair<int,int> >t=(*it).second;
vector<pair<int,int> >v;
int sz=t.size();
int num=0;
//cout<<it->first<<" "<<sz<<endl;
sort(t.begin(),t.end(),cmp);
pair<int,int> p=t[0];
v.push_back(make_pair(t[0].first,t[0].second));
num++;
for(int i=1;i<sz;i++){
if(t[i].first>p.second){
v.push_back(make_pair(t[i].first,t[i].second));
num++;
p=t[i];
}
}
if(ans<num){
ans=num;
as=v;
}
}
printf("%d\n",ans );
for(int i=0;i<(int)as.size();i++)
printf("%d %d\n",as[i].first,as[i].second);
return 0;
}
其实感觉思路还是蛮简单的。
这道题需要注意以下几点,也是需要学习的几处:
- map<int,vector<pair<int,int> > >ma;的相关用法。
- vector竟然能直接地址赋值!!!(恕我才疏学浅,c++基本不会)
- 再强调一遍,对于map的迭代器的用法。