当前位置: 首页 > 工具软件 > Frets On Fire > 使用案例 >

Codeforces Global Round 2 D题 Frets On Fire 差分,前缀和,二分

鲜于喜
2023-12-01

题目链接

http://codeforces.com/contests

题意:输入n个数字,代表n个序列,每个序列的首位为s[i],公差为1,接下来输入询问次数,每次询问[l,r]区间内所有序列总共不同的数字的个数。

题解

转发队友cy的题解

https://blog.csdn.net/chenyume/article/details/89065578

时间复杂度是O(nlogn+qlogn)

代码基本上一样,就是二分查找的时候可以直接调用c++,STL的upper_bound()函数。

代码

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=1e5+5;
ll a[maxn];
ll d[maxn];     //差分数组
ll sum[maxn];   //差分数组的前缀和
int n,m;
void init(){
    sort(a,a+n); 
    m=unique(a,a+n)-a;      //去重
    for(int i=1;i<m;i++){
        d[i]=a[i]-a[i-1];
    }
    sort(d+1,d+m);
    for(int i=1;i<m;i++){
        sum[i]+=sum[i-1]+d[i];
    }
}
int main(){
    cin>>n;
    for(int i=0;i<n;i++){
        cin>>a[i];
    }
    init();
    int q;
    cin>>q;
    while(q--){
        ll l,r;
        cin>>l>>r;
        ll x=r-l+1;
        ll index=upper_bound(d+1,d+m,x)-d;  //找差分数组第一个大于x的位置
        ll ans=sum[index-1]+x*(m-index+1);
        cout<<ans<<endl;
    }
    return 0;
}

 

 类似资料: