1. 程式人生 > >牛客練習賽7 E 珂朵莉的數列

牛客練習賽7 E 珂朵莉的數列

arc scanner print amp tar [] next ++ implement

珂朵莉的數列

思路:

樹狀數組+高精度

離散化不知道哪裏寫錯了,一直wa,最後用二分寫的離散化

哪位路過大神可以幫我看看原來的那個離散化錯在哪裏啊

通過代碼:

import java.math.BigInteger;
import java.util.*;
import java.util.Scanner;
  
class node
{
    int x;
    int id;
}
class cmp implements Comparator<node>
{
    public int compare(node A,node B)
    {
        
if(A.x<B.x)return -1; else if(A.x==B.x)return 0; else return 1; } } public class Main{ public final static int N=(int)1e6+9; public static int n; public static int a[]=new int[N]; public static long b[]=new long[N]; public static long bit[]=new long
[N]; public static long sum(int x) { long ret=0; while(x>0) { ret+=bit[x]; x-=x&(-x); } return ret; } public static void add(int x,long a) { while(x<=n) { bit[x]+=a; x
+=x&(-x); } } public static void main(String[] args) { Scanner reader = new Scanner(System.in); n=reader.nextInt(); for(int i=1;i<=n;i++) { a[i]=reader.nextInt(); b[i]=a[i]; } Arrays.sort(b,1,n+1); for(int i = 1; i <= n; i++) { a[i] = Arrays.binarySearch(b, 1 ,n + 1,a[i]); } BigInteger ans=BigInteger.valueOf(0); for(int i=1;i<=n;i++) { ans=ans.add(BigInteger.valueOf((sum(n)-sum(a[i]))*(n-i+1))); add(a[i],i); } System.out.println(ans); } }

錯誤代碼:

import java.math.BigInteger;
import java.util.*;
import java.util.Scanner;
  
class node
{
    int x;
    int id;
}
class cmp implements Comparator<node>
{
    public int compare(node A,node B)
    {
        if(A.x<B.x)return -1;
        else if(A.x==B.x)return 0;
        else return 1;
    }
}
public class Main{
    public final static int N=(int)1e6+9;
    public static int n;
    public static node a[]=new node[N];
    public static int b[]=new int[N];
    public static long bit[]=new long[N];
    public static long sum(int x)
    {
        long ret=0;
        while(x>0)
        {
            ret+=bit[x];
            x-=x&(-x);
        }
        return ret;
    }
    public static void add(int x,long a)
    {
        while(x<=n)
        {
            bit[x]+=a;
            x+=x&(-x);
        }
    }
    public static void main(String[] args) {
            Scanner reader = new Scanner(System.in);
            n=reader.nextInt();
            for(int i=1;i<=n;i++)
            {
                a[i]=new node();
                a[i].x=reader.nextInt();
                a[i].id=i;
            }
            //
            Arrays.sort(a,1,n+1,new cmp());
              
            int cnt=1;
            a[0]=new node();
            a[0].x=-1;
            for(int i=1;i<=n;i++)
            {
                if(a[i].x!=a[i-1].x)b[a[i].id]=++cnt;
                else b[a[i].id]=cnt;
            }
            //for(int i=1;i<=n;i++)System.out.println(b[i]);
            BigInteger ans=BigInteger.valueOf(0);
            for(int i=1;i<=n;i++)
            {
                ans=ans.add(BigInteger.valueOf((sum(cnt)-sum(b[i]))*(n-i+1)));
                add(b[i],i);
            }
            System.out.println(ans);
        }
}

牛客練習賽7 E 珂朵莉的數列