《算法竞赛进阶指南》0x05 排序 离散化

Posted 2020-12-12 randy-lo

题目链接：http://codeforces.com/contest/670/problem/C

一个电影有影视语言以及字幕语言，有m部电影以及n个人，每个人只会一种语言，语言的编号是一个int数，如果一个人能听懂影视语言就会非常开心，如果他听不懂影视语言但是听得懂字幕语言他将会比较高兴，如果他两个语言都听不懂就会不高兴，问他们应该去看哪部电影才能使得非常开心的人最多？如果非常开心的人的数量是一样的就要选择一部电影使得比较开心的人是最多的。

题解：由于语言是int范围的，对于每场电影，只要统计n个人中会这两种语言的人数就能知道非常开心的人数以及比较开心的人数，此时可以通过离散化的方法处理之后对每种语言会的人数进行统计。

代码如下：

 

#include<bits/stdc++.h>
using namespace std;
typedef unsigned int ui;
typedef long long ll;
typedef unsigned long long ull;
#define pf printf
#define mem(a,b) memset(a,b,sizeof(a))
#define prime1 1e9+7
#define prime2 1e9+9
#define pi 3.14159265
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
#define scand(x) scanf("%llf",&x) 
#define f(i,a,b) for(int i=a;i<=b;i++)
#define scan(a) scanf("%d",&a)
#define mp(a,b) make_pair((a),(b))
#define P pair<int,int>
#define dbg(args) cout<<#args<<":"<<args<<endl;
#define inf 0x7ffffff
inline int read(){
    int ans=0,w=1;
    char ch=getchar();
    while(!isdigit(ch)){if(ch==‘-‘)w=-1;ch=getchar();}
    while(isdigit(ch))ans=(ans<<3)+(ans<<1)+ch-‘0‘,ch=getchar();
    return ans*w;
}
int n,m,t;
const int maxn=2e5+10;
int language[maxn*3];//最多有2*m+n中语言 
int a[maxn],x[maxn],y[maxn],z[maxn];
int ans[maxn*3];
const ll mod=10000;
int k;
int find(int f){
    return lower_bound(a+1,a+k+1,f)-a;//获得地第一个大于等于f的位置的下标 
} 
int main()
{
//    freopen("input.txt","r",stdin);
//    freopen("output.txt","w",stdout);
        cin>>n;
        int tot=0;
        f(i,1,n)z[i]=read(),a[++tot]=z[i];//每个人会的语言种类 
        cin>>m;
        f(i,1,m)x[i]=read(),a[++tot]=x[i];//读取电影的两种语言 
        f(i,1,m)y[i]=read(),a[++tot]=y[i];
        sort(a+1,a+tot+1);
        k=unique(a+1,a+tot+1)-(a+1);//统计数量，unique函数返回的是无重复的下一个位置
         mem(ans,0);
         f(i,1,n)ans[find(z[i])]++;
         int ret=0,ret2=0;
         int ans1,ans2,ans0=1;//如果ans0没有被赋值的话就说明语言是没有交集的 
         f(i,1,m){
             ans1=ans[find(x[i])],ans2=ans[find(y[i])];
             if(ans1>ret || (ans1==ret && ans2>ret2)){
                 ans0=i;//选择的电影数
                ret=ans1;
                ret2=ans2;    
            }
         }
         cout<<ans0<<endl;
}