Codeforces Round #345 (Div. 1)

A. Watchmen

只有在和坐标轴平行的线上的两点，两种距离相等。对于重合的点，容斥一下。

#include <bits/stdc++.h>
#include <unordered_map> 

using namespace std;

#define ll long long

unordered_map<int,ll> xx;
unordered_map<int,ll> yy;

struct point{
    int x;
    int y;
    point(int x,int y):x(x),y(y){

    }
    point(){

    }
};

map<pair<int,int> ,ll> pts;

int main(){

    int n;

    cin>>n;

    for(int i=1;i<=n;i++){
        int x,y;
        scanf("%d%d",&x,&y);
        xx[x]++;
        yy[y]++;
        pts[make_pair(x,y)]++;
    }

    ll ans = 0;

    for(auto it=xx.begin();it!=xx.end();it++){
        ans+= it->second*(it->second-1)/2;
    }

    for(auto it=yy.begin();it!=yy.end();it++){
        ans+= it->second*(it->second-1)/2;
    }

    for(auto it=pts.begin();it!=pts.end();it++){
        ans-= it->second*(it->second-1)/2;
    }

    cout<<ans<<endl;

    return 0;
}

B. Image Preview

分析一下可以知道，如果改变方向，肯定只会改变一次。对于先往左和先往右的情况，枚举转向的位置，二分搜一下。

#include <bits/stdc++.h>
#include <unordered_map> 

using namespace std;

#define ll long long


int n,a,b,t;

int lr[500010];
int rl[500010];

char ori[500010];

int bslr(int k){
    int curt = t;
    curt-=lr[k];
    if(curt<0)return 0;
    curt-=a*(k-1);
    if(curt<=0)return k;


    int r = n+1;
    int l = k+1;
    int finalPos=n+1;
    while(l<=r){
        int mid = (r+l)>>1;
        if(rl[mid]>curt){
            l=mid+1;
        }else{
            finalPos=mid;
            r=mid-1;
        }
    }
    return k+(n+1-finalPos);
}

int ans = 0;

void solve(){
    lr[0] =-a;
    for(int i=1;i<=n;i++){
        lr[i]=lr[i-1];
        lr[i] += a+1;
        if(ori[i]=='w'){
            lr[i]+=b;
        }
    }

    rl[n+1]=0;
    for(int i=n;i>1;i--){
        rl[i]=rl[i+1];
        rl[i] += a+1;
        if(ori[i]=='w'){
            rl[i]+=b;
        }
    }
    rl[n+1]=0;


    for(int i=1;i<=n;i++){
        ans = max(ans,bslr(i));
    }
}

int main(){
    cin>>n>>a>>b>>t;

    scanf("%s",ori+1);

    solve();
    reverse(ori+2,ori+n+1);
    solve();

    cout<<ans<<endl;

    return 0;
}

C. Table Compression

用并查集合并必须一样的点，然后对值排序，从小到大处理。每次处理到新的点，要保证它比同行，同列中要小的点大，维护所在集合即可。比赛时用unordered_map乱搞，居然完全没有想到并查集。

#include <bits/stdc++.h>
#include <unordered_map> 

using namespace std;

#define ll long long

int tab[1000010];

int ans[1000010];

const int maxsz = 1000010;

int rowMax[maxsz];
int colMax[maxsz];

int rowMaxId[maxsz];
int colMaxId[maxsz];


struct node{
    int val;
    int x,y;
    node(int val,int x,int y):val(val),x(x),y(y){
    }
    node(){
    }
    bool operator<(const node &other)const{
        return val<other.val;
    }
}nds[1000010];


int p[1000010];

int find(int x){
    if(p[x]==x)return x;
    int px = find(p[x]);
    return p[x]=px;
}

void Union(int a,int b){
    int pa = find(a);
    int pb = find(b);
    if(pa!=pb){
        p[pb]=pa;
    }
}


int main(){
    int n,m;
    cin>>n>>m;

    int k=1;
    for(int i=0;i<n;i++){
        for(int j=0;j<m;j++){
            int num;
            scanf("%d",&num);
            nds[k]=node(num,i,j);
            tab[k++]=num;
        }
    }

    int nm=n*m;
    for(int i=0;i<=nm;i++){
        p[i] = i;
    }

    for(int i=0;i<n;i++){
        unordered_map<int,int> mp;
        for(int j=0;j<m;j++){
            int k = i*m+j+1;
            if(mp.count(tab[k])){
                Union(k,mp[tab[k]]);
            }else{
                mp[tab[k]] = k;
            }
        }
    }
    for(int j=0;j<m;j++){
        unordered_map<int,int> mp;
        for(int i=0;i<n;i++){
            int k = i*m+j+1;
            if(mp.count(tab[k])){
                Union(k,mp[tab[k]]);
            }else{
                mp[tab[k]] = k;
            }
        }
    }

    sort(nds+1,nds+nm+1);
    for(int i=1;i<=nm;i++){
        int val = nds[i].val;
        int x = nds[i].x;
        int y = nds[i].y;

        int k = find(x*m+y+1);
        if(rowMax[x]<val){
            ans[k] = max(ans[k],ans[rowMaxId[x]]+1); 
            rowMax[x]=val;
            rowMaxId[x] = k;
        }
        if(colMax[y]<val){
            ans[k] = max(ans[k],ans[colMaxId[y]]+1);
            colMax[y]=val;
            colMaxId[y] = k;
        }
    }

    for(int i=0;i<n;i++){
        for(int j=0;j<m;j++){
            int aa = ans[find(i*m+j+1)];
            printf("%d ",aa);
        }
        printf("\n");
    }

    return 0;
}