稀疏向量 (CCF20200602)

稀疏向量 (CCF20200602)_第1张图片
稀疏向量 (CCF20200602)_第2张图片

稀疏向量 (CCF20200602)_第3张图片
![在这里插入图片描述](https://img-blog.csdnimg.cn/20200727121855472.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQzNzQ5MjYy,size_16,color_FFFFFF,t_70

题记

像拉拉链一样把两个数组都刷一遍,注意内部往后刷的时候不要越界。另外保存答案的ans要用long long。本题其实默认是下标升序,如果不放心可以写一个cmp用sort排一下。
代码如下

#include 
#include 
using namespace std;

const int Maxn=1000000;

//输入
int n,a,b;

struct vertex{
    int index;
    int value;
};

bool cmp(vertex&a,vertex&b){
    return a.index<=b.index;
}

vertex u[Maxn],v[Maxn];

void solve(){
    int i=0,j=0;
    long long ans=0;
    sort(u,u+a,cmp);
    sort(v,v+b,cmp);
    while(i<a&&j<b){
        if(u[i].index<v[j].index){
            while(u[i].index<v[j].index&&i<a)i++;
        }
        else if (u[i].index>v[j].index&&j<b) {
            while(u[i].index>v[j].index)j++;
        }

        if (u[i].index==v[j].index) {
            ans+=u[i].value*v[j].value;
            i++,j++;
        }

     }
    printf("%lld\n",ans);
}

int main()
{
    scanf("%d %d %d",&n,&a,&b);
    for(int i=0;i<a;i++)
        scanf("%d %d",&u[i].index,&u[i].value);
    for(int i=0;i<b;i++)
        scanf("%d %d",&v[i].index,&v[i].value);

    solve();
    
    return 0;
}

你可能感兴趣的:(#,CCF202006)