Educational Codeforces Round #41 (D&E)

ABC就是暴力暴力暴力哇……

D. Pair of Lines: 给出平面上n个点的坐标，问是否存在两条直线使得这n个点中的每一个都至少位于一条直线上。

两条直线，首先要确定其中一条直线，然后将不在这条直线上的所有点取出来判断它们是否共线。

前三个点必定可以确定一条直线，直接枚举三种组合方式。

n<=4时直接输出Yes。

#include<cstdio>
#include<cstring>
#include<queue>
#include<vector>
#include<map>
#include<cmath>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#define maxn 100050
#define INF 0x3f3f3f3f
#define eps 1e-10
using namespace std;
typedef long long ll;

ll x[maxn],y[maxn];
int n;

bool aline(int a,int b,int c)
{
    return ((x[b]-x[a])*(y[c]-y[b])==(x[c]-x[b])*(y[b]-y[a]));
}

bool judge(int u,int v)
{
    vector<int>ve;
    for(int i=1;i<=n;i++)
    {
        if(!aline(u,v,i))
            ve.push_back(i);
    }
    int len=ve.size();
    if(len<3)return true;
    for(int i=0;i<len;i++)
    {
        if(!aline(ve[0],ve[1],ve[i]))
            return false;
    }
    return true;
}

int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        scanf("%I64d%I64d",&x[i],&y[i]);
    }
    if(n<=4)
    {
        printf("Yes\n");
        return 0;
    }
    if(judge(1,2)||judge(1,3)||judge(2,3))
        printf("Yes\n");
    else printf("No\n");
    return 0;
}

E. Tufurama: 给出一个数列a，问对于每一个i，有多少个j>i满足a[j]>=i，求其总和。

一开始开了一个普通的一维线段树然后眉头一皱发现事情并不简单，在某些情况下时间复杂度会接近O(n^2)。

原因是每次查询的时候可能会遍历到树的底部，于是就需要对线段树进行合并。

线段树的每一个节点都是一个vector，在建树的时候对每一个区间进行合并，将区间内所有的a[i]按升序排列在vector中。

在查询中遍历到可行区间时，要找出区间中有多少个数不小于给定值只要二分即可。

#include<cstdio>
#include<cstring>
#include<queue>
#include<vector>
#include<map>
#include<cmath>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#define maxn 200050
#define INF 0x3f3f3f3f
#define eps 1e-8
using namespace std;
typedef long long ll;

int n;
int a[maxn];
vector<int>tree[maxn*4];

void Merge(int root,int l,int r)
{
    int lp=0,rp=0;
    int ln=tree[l].size(),rn=tree[r].size();
    while(lp<ln&&rp<rn)
    {
        if(tree[l][lp]<tree[r][rp])
        {
            tree[root].push_back(tree[l][lp]);
            lp++;
        }
        else
        {
            tree[root].push_back(tree[r][rp]);
            rp++;
        }
    }
    while(lp<ln)
    {
        tree[root].push_back(tree[l][lp]);
        lp++;
    }
    while(rp<rn)
    {
        tree[root].push_back(tree[r][rp]);
        rp++;
    }
}

void build(int root,int l,int r)
{
    if(l==r)
    {
        tree[root].push_back(a[l]);
        return;
    }
    int mid=(l+r)/2;
    build(root*2,l,mid);
    build(root*2+1,mid+1,r);
    Merge(root,root*2,root*2+1);
}

ll query(int root,int l,int r,int aiml,int aimr,int tmp)
{
    if(l>aimr||r<aiml)
        return 0;
    if(l>=aiml&&r<=aimr)
    {
        int len=tree[root].size();
        if(tree[root][len-1]<tmp)
            return 0;
        int pos=lower_bound(tree[root].begin(),tree[root].end(),tmp)-tree[root].begin();
        return len-pos;
    }
    int mid=(l+r)/2;
    return query(root*2,l,mid,aiml,aimr,tmp)+query(root*2+1,mid+1,r,aiml,aimr,tmp);
}

int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);
    }
    build(1,1,n);
    ll ans=0;
    for(int i=1;i<=n;i++)
    {
        if(a[i]<=i)continue;
        ans+=query(1,1,n,i+1,a[i],i);
    }
    printf("%I64d\n",ans);
    return 0;
}