E. Count The Rectangles [思维+优先队列+树状数组]

传送门
题意:给出n(n<=5000)条水平和垂直的线段,计算共形成多少个矩形

题解:枚举矩形的上下边即枚举水平线段,显然如果有x个垂直线段与这两条边同时相交那么就形成了x*(x-1)/2个矩形,如果直接暴力枚举垂直线段,那么复杂度为O(n^3),复杂度爆表是因为垂直线段这样被重复计算贡献,思考能不能快速地计算贡献,对于纵坐标为(y1,y2)的垂直线段,其能产生贡献的区间就是下边界为>=y1上边界<=y2,那么可以在枚举上下边界时利用优先队列维护一个满足条件的序列,通过树状数组快速更改新加入的垂直线段的贡献或者删除不满足条件的垂直线段的贡献,复杂度变成了O( n^2 *logn)

#include<bits/stdc++.h>

using namespace std;
typedef long long ll;
#define debug(x) cout<<#x<<" is "<<x<<endl;

const int maxn=1e5+5;

struct edge{
   
   
    int x;
    int y;
    int z;
    int t;
    int id;
    bool operator<(const struct edge &aa)const{
   
   
        return y>aa.y;
    }
}h[maxn],v[maxn];

bool cmp(struct edge aa,struct edge bb){
   
   
    return aa.x<bb.x;
}

int maxx,tim,wq[maxn];
ll sum[maxn];

void update(int x,int y){
   
   
    while(x<=maxx){
   
   
        if(wq[x]!=tim){
   
   
            wq[x
# P7530 [USACO21OPEN] United Cows of Farmer John P ## 题目描述 农夫约翰合牛国(The United Cows of Farmer John,UCFJ)将要选派一个代表队参加国际牛学奥林匹克(International bOvine olympIad,IOI)。 有 $N$ 头奶牛参加了代表队选拔。她们站成一行,奶牛 $i$ 的品种为 $b_i$。 代表队将会由包含至少三头奶牛的连续区间组成——也就是说,对于满足 $1\le l<r\le N$ 且 $r−l\ge 2$ 的奶牛 $l\dots r$。选定区间内的三头奶牛将会被指定为领队。出于法律原因,最边上的两头奶牛必须是领队。此外,为了避免种内冲突,每一名领队都必须与代表队的其他成员(包括领队)品种不同。 请帮助 UCFJ 求出(由于纳税原因)他们可以选派参加 IOI 的代表队的方法数。如果两个代表队拥有不同的成员或不同的领队,则被认为是不同的。 ## 输入格式 输入的第一行包含 $N$。 第二行包含 $N$ 个整数 $b_1,b_2,\dots,b_N$,均在范围 $[1,N]$ 之间。 ## 输出格式 输出一行一个整数表示可能的代表队的数量。 ## 输入输出样例 #1 ### 输入 #1 ``` 7 1 2 3 4 3 2 5 ``` ### 输出 #1 ``` 9 ``` ## 说明/提示 #### 样例解释 每一代表队对应以下的一组领队: $$(1,2,3),(1,2,4),(1,3,4),(1,4,7),(2,3,4),(4,5,6),(4,5,7),(4,6,7),(5,6,7).$$ #### 数据范围与约定 $1\le N\le 2\times 10^5$ #include <bits/stdc++.h> #define ll long long using namespace std; const ll N=2e5; ll n; ll a[N+5]; ll pre[N+5],pos[N+5]; ll tr[4*N+5],lz[4*N+5]; ll ans; void update(ll p){ tr[p]=tr[2*p]+tr[2*p+1]; return; } void pushdown(ll p,ll l,ll r){ if(!lz[p]) return; ll mid=(l+r)>>1; tr[2*p]+=(mid-l+1)*lz[p]; tr[2*p+1]+=(r-mid)*lz[p]; lz[2*p]+=lz[p]; lz[2*p+1]+=lz[p]; lz[p]=0; return; } void change(ll p,ll l,ll r,ll x,ll y,ll z){ if(x>y) return; if(x<=l && r<=y){ tr[p]+=(r-l+1)*z; lz[p]+=z; return; } pushdown(p,l,r); ll mid=(l+r)>>1; if(y<=mid) change(2*p,l,mid,x,y,z); else if(x>=mid+1) change(2*p+1,mid+1,r,x,y,z); else{ change(2*p,l,mid,x,mid,z); change(2*p+1,mid+1,r,mid+1,y,z); } update(p); return; } ll searchh(ll p,ll l,ll r,ll x,ll y){ if(x>y) return 0; if(l==x && r==y) return tr[p]; pushdown(p,l,r); ll mid=(l+r)>>1; if(y<=mid) return searchh(2*p,l,mid,x,y); else if(x>=mid+1) return searchh(2*p+1,mid+1,r,x,y); else return searchh(2*p,l,mid,x,mid)+searchh(2*p+1,mid+1,r,mid+1,y); } int main(){ scanf("%lld",&n); for(ll i=1;i<=n;i++){ scanf("%lld",&a[i]); pre[i]=pos[a[i]]; pos[a[i]]=i; } for(ll i=1;i<=n;i++){ ans+=searchh(1,1,n,pre[i]+1,i-1); if(pre[i]){ change(1,1,n,pre[i],pre[i],-searchh(1,1,n,pre[i],pre[i])); change(1,1,n,pre[i]+1,i-1,1); } else change(1,1,n,1,i-1,1); if(pre[pre[i]]) change(1,1,n,pre[pre[i]]+1,pre[i]-1,-1); else if(pre[i]) change(1,1,n,1,pre[i]-1,-1); // cout<<searchh(1,1,n,pre[i]+1,i-1)<<endl; // for(ll j=1;j<=n;j++){ // cout<<searchh(1,1,n,j,j)<<" "; // } // cout<<endl; } printf("%lld",ans); return 0; } /* 7 1 2 3 4 3 2 5 */
08-12
评论 1
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值