Coder-Strike 2014 - Finals (online edition, Div. 2) D.Bug in Code

一道題意蠻好理解的題......

想法:記錄下每個人被懷疑的次數,並且以此排序,從最左開始,對於第i個人,在[i+1,n]區間二分出和它相加和>=p的最小下標j,答案就加上n-j+1,但顯然這樣有重複計算,因爲可能選中的兩個人被同一個人懷疑,所以要去重,用一個map來記錄(i,j)被幾個人同時懷疑,就可以知道這兩個人能不能被選......


/*
 * File Name: cf.cpp
 * Create Time: 19:54:02,星期二, 22 四月 2014.
 * Author: qqspeed
 */

#include<algorithm>
#include<bitset>
#include<deque>
#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<limits>
#include<list>
#include<map>
#include<queue>
#include<set>
#include<sstream>
#include<stack>
#include<vector>
#include<complex>
using namespace std;

#define FOR(i,b,e) for(__typeof(b)i=(b);i!=(e);i++)
#define FOR_EACH(i,c) for(__typeof((c).begin())i=(c).begin();i!=(c).end();i++)
#define rep(i,s,t) for(int i=s;i<t;i++)
#define INF ~0U>>2
#define L t<<1
#define R t<<1|1
#define PI acos(-1.0)
#define clr(a) memset(a,0,sizeof(a))
typedef complex<double>Com;
typedef pair<int,int> ii;
typedef long long ll;
typedef unsigned long long ull;

inline void DATA(char *s,char *d){
    freopen(s,"r",stdin);
    freopen(d,"w",stdout);
}

inline int input(){
    int ret=0;bool isN=0;char c=getchar();
    while(c<'0' || c>'9'){
        if(c=='-') isN=1;
        c=getchar();
    }
    while(c>='0' && c<='9'){
        ret=ret*10+c-'0';c=getchar();
    }
    return isN?-ret:ret;
}

inline void output(int x){
    if(x<0){
        putchar('-');x=-x;
    }
    int len=0,data[10];
    while(x){
        data[len++]=x%10;x/=10;
    }
    if(!len)    data[len++]=0;
    while(len--)
        putchar(data[len]+48);
    putchar('\n');
}

#define N 300005
map<ii,int>mp;
int n,p;
int a[N],b[N];
struct node{
    int id,val;
}root[N];
bool cmp(node i,node j){
    return i.val<j.val||(i.val==j.val&&i.id<j.id);
}
ll ans;
inline int BS(int s,int t,int v){
    int mid;
    int p=t+1;
    while(s<=t){
        mid=(s+t)>>1;
        if(root[mid].val>=v) p=min(p,mid); 
        if(root[mid].val<v) s=mid+1;
        else t=mid-1;
    }
    return p;
}
int cnt[N];
int main(){
    ans=0;
    clr(cnt);
    mp.clear();
    n=input(),p=input();
    rep(i,1,n+1) root[i].id=i,root[i].val=0;
    rep(i,1,n+1){
        a[i]=input(),b[i]=input();
        if(a[i]>b[i]) swap(a[i],b[i]);
        mp[ii(a[i],b[i])]++;
        root[a[i]].val++;
        root[b[i]].val++;
    }
    for(map<ii,int>::iterator it=mp.begin();it!=mp.end();it++){
        {
            int x=(*it).first.first;
            int y=(*it).first.second;
            if(root[x].val+root[y].val>=p && root[x].val+root[y].val-(*it).second <p ) ans--;
        }
    }
    sort(root+1,root+n+1,cmp);
    rep(i,1,n){
        {
            int r=p-root[i].val;
            int s=BS(i+1,n,r);
// output(s);
            ans+=(n-s+1)*1LL;
        }
    }
    printf("%lld\n",ans);
    return 0;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值