Codeforces Round #547 (Div. 3) F 贪心 + 离散化

Posted 2021-11-24 virtu0s0

https://codeforces.com/contest/1141/problem/F2

题意

一个大小为n的数组a[],问最多有多少个不相交的区间和相等

题解

• 离散化用值来做,贪心选择较前的区间

代码

#include<bits/stdc++.h>
#define M 5000005
#define ll long long 
#define pb push_back
using namespace std;
struct N{int l,r;N(int l=0,int r=0):l(l),r(r){}};
int n,i,j,p,ans,cnt,u,r,sz=0;
ll a[2005],b[M];
vector<N>g[M];

int main(){
    cin>>n;
    for(i=1;i<=n;i++){
        scanf("%lld",&a[i]);a[i]+=a[i-1];
        for(j=i;j>=1;j--){
            b[sz++]=a[i]-a[j-1];
        }
    }
    sort(b,b+sz);sz=unique(b,b+sz)-b;
    for(i=1;i<=n;i++){
        for(j=1;j<=i;j++){
            p=lower_bound(b,b+sz,a[i]-a[j-1])-b;
            g[p].pb({j,i});
        }
    }
    ans=1;
    for(i=0;i<sz;i++){
        r=g[i][0].r;cnt=1;
        for(j=1;j<g[i].size();j++){
            if(g[i][j].l>r){
                cnt++;r=g[i][j].r;
            }
        }
        if(cnt>=ans){
            ans=cnt;u=i;
        }
    }
    printf("%d\n%d %d\n",ans,g[u][0].l,g[u][0].r);
    r=g[u][0].r;
    for(i=1;i<g[u].size();i++){
        if(g[u][i].l>r){
            printf("%d %d\n",g[u][i].l,g[u][i].r);
            r=g[u][i].r;
        }
    }
}