洛谷P2205 [USACO13JAN]画栅栏Painting the Fence

洛谷P2205 [USACO13JAN]画栅栏Painting the Fence
这是我的做法，我用了离散化+二分，比较慢

#include <bits/stdc++.h> 
#define For(i,j,k) for(int i=j;i<=k;i++) 
#define int long long 
#define LL long long 
using namespace std ; 

const int N = 100011 ; 
struct node{
    int l,r,del ; 
}cnt[N*2];
int n,x1,x2,v,dir,num,t,K,tot ; 
LL sum ; 
int L[N],R[N],tmp[2*N],point[2*N] ; 
char s[4] ; 

inline int read() 
{
    int x = 0 , f = 1 ; 
    char ch = getchar() ; 
    while(ch<'0'||ch>'9') { if(ch=='-') f = -1 ; ch = getchar() ; } 
    while(ch>='0'&&ch<='9') { x = x * 10+ch-48 ; ch = getchar() ; } 
    return x * f ;  
}

inline int erfen(int x)   // 在 point 数组中找 x 
{
    int l = 1,r = t ; 
    while(l<r) {
        int mid=(l+r)/2 ; 
        if(point[mid]>=x) 
            r=mid ; 
        else 
            l=mid+1 ;         
    }
    return r ; 
}

signed main() 
{
    n = read() ; K = read() ; 
    x1 = 0 ; x2 = 0 ; 
    For(i,1,n) {
        v = read() ; 
        scanf("%s",s+1) ; 
        if(s[1]=='L') x2-=v ; 
        else x2+=v; 
        L[i]=x1 ; R[i]=x2 ; 
        x1=x2 ; 
    } 
    For(i,1,n) if(L[i]>R[i]) swap(L[i],R[i]) ; 
    For(i,1,n) tmp[++tot]=L[i],tmp[++tot]=R[i] ; 
    sort(tmp+1,tmp+tot+1) ; 
    
    point[++t] = tmp[1] ; 
    For(i,2,tot) 
        if(tmp[i]!=tmp[i-1]) point[++t] = tmp[i] ; 
    For(i,1,t-1) {
        cnt[i].l = point[i] ;  
        cnt[i].r = point[i+1] ; 
        cnt[i].del = 0 ; 
    }
    
    For(i,1,n) {
        int l = erfen(L[i]) ; 
        cnt[l].del++ ; 
        int r = erfen(R[i])-1 ;    
        cnt[r+1].del-- ; 
    }
    For(i,1,t-1) {
        num+=cnt[i].del ; 
        if(num>=K) sum+=cnt[i].r-cnt[i].l ; 
    }
    printf("%lld
",sum) ; 
    return 0 ; 
}

其实差分就行了，不用二分
虽然时间复杂度也是 nlogn
但是因为不用而二分了，所以快了不少
大概快了300ms
另外也可以用map直接水

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<cstring>
int i,j,k,m,n,x,y,wz;
struct ppp{
    int wz,lr;
}a[200001];
bool cmp(ppp x,ppp y)
{
    return x.wz<y.wz;
}
char c;
using namespace std;
int main()
{
    scanf("%d %d",&n,&m);
    for(i=1;i<=n;i++)
    {
        scanf("%d %c",&x,&c);
        a[i*2-1].wz=wz;
        if(c=='R')wz+=x;else wz-=x;
        a[i*2].wz=wz;
        if(a[i*2-1].wz<a[i*2].wz)
        a[i*2-1].lr=0,a[i*2].lr=1;
        else 
        a[i*2-1].lr=1,a[i*2].lr=0;
    }//存端点信息而不是线段信息

    sort(a+1,a+2*n+1,cmp);
    int ans=0;
    for(i=1;i<=n*2;i++)
    {
        if(k>=m)ans+=(a[i].wz-a[i-1].wz);
        if(a[i].lr==0)k++;else k--;
    }
    cout<<ans;
    return 0;
}