Question

在问题中，我解析输入（整数）并同时检查它是否存在于数据结构中，如果不存在则添加它。

输入是 - 由大小＆gt; = 1和＆lt; = 1000000

的空格分隔的2个整数

我尝试使用HashMap，TreeMap（put()和containsValue()方法） - 但似乎他们花了太多时间。（10个测试用例中有5个超过了时间限制）

使用ArrayList（add（）和contains（）方法时） - （10个测试用例中有4个超出了时间限制）

这些操作将在第二个for循环内执行，在if条件内。

迭代可能如下变化： -

第一个循环 - 1到10

第二个循环 - 1到100000

所以我猜测第二次循环中的高阶迭代超过了时间限制。

还有其他方法可以在较短的时间内执行此任务。

问题：

一个和尚遇到N个池塘，每个池塘都有一个食物项目（输入1）和一个口袋妖怪（输入2）。

如果类型匹配，僧侣可以将第i个池塘的物品喂给池塘的口袋妖怪。在离开之前，和尚可能不得不随身携带一些食物，以便喂养所有的小宠物。帮助他找到必须携带的物品数量，以便能够安全地穿过这片土地。

解释

在First Pond，他获得了type1的项目并将其提供给了type1的Pokemon。

在Second Pond，他获得了类型2的项目并将其提供给类型2的口袋妖怪。

在Third Pond，他获得了类型3的项目，但是口袋妖怪是类型4。因此，他必须携带4型食品。

在第四池，他得到了类型4的物品。他已经有一个3型物品并将它喂给口袋妖怪。

在第五池塘，他得到了2型物品。他已经有一个4型物品并将它喂给这个池塘的口袋妖怪

class TestClass {
public static void main(String args[] ) throws Exception {

    BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

    int T = Integer.parseInt(br.readLine());
    if(T<=10 && T>=1)   {
    for (int i = 0; i < T; i++) {
       int count=0;
       int numOfPonds = Integer.parseInt(br.readLine());
        if(numOfPonds<=100000 && numOfPonds>=1){  
       String[] str ;
       Map m = new HashMap();
        //List l = new ArrayList();

        for(int j=0 ; j< numOfPonds ;j++)
                {   
                    str = br.readLine().split(" ");
                    int foodType = Integer.parseInt(str[0]);
                    int PokeType = Integer.parseInt(str[1]);
                    if(foodType<=1000000 && PokeType<=1000000 && foodType>=1 && PokeType>=1 && foodType != PokeType){

                        m.put(j,foodType);

                    //l.add(foodType);

                        //if(!(l.contains(PokeType)))
                    if(!(m.containsValue(PokeType)))                        
                                    count++;

                    //else if(l.contains(PokeType))
                    else if(m.containsValue(PokeType))
                        {
                            m.values().remove(PokeType);
                            //  l.remove(Integer.valueOf(PokeType));
                            }

                    }
                }
        }
       System.out.println(count);
      }
    }
    }
}

Answer 1

以下是在时间限制内通过所有测试用例的代码。

如 Codebender 和 JimN 所述，我实施了containsKey()方法，证明速度比containsValue()快。

另外，对于重复项，增加并更改了Map中的值。

class TestClass {
public static void main(String args[] ) throws Exception {

BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

int T = Integer.parseInt(br.readLine());
if(T<=10 && T>=1)   {
for (int i = 0; i < T; i++) {
   int count=0;
   int numOfPonds = Integer.parseInt(br.readLine());
    if(numOfPonds<=100000 && numOfPonds>=1){  
   String[] str ;

Map<Integer,Integer> map = new HashMap<Integer,Integer>();
    for(int j=0 ; j< numOfPonds ;j++)
            {   
                str = br.readLine().split(" ");
                int foodType = Integer.parseInt(str[0]);
                int PokeType = Integer.parseInt(str[1]);

                if(foodType<=1000000 && PokeType<=1000000 && foodType>=1 && PokeType>=1 && foodType != PokeType){

                if(map.containsKey(foodType))
                {
                    int x = map.get(Integer.valueOf(foodType));
                    x++;
                    map.put(foodType,x);
                }
                else
                {   map.put(foodType,1);
                }

                if(!(map.containsKey(PokeType)))                        
                                count++;

                else 
                    {   int x = map.get(Integer.valueOf(PokeType));
                        x--;

                        if(x==0)
                        map.remove(PokeType);

                        else
                        map.put(PokeType,x);

                        }

                }
            }
     }
   System.out.println(count);
  }
}
}
}

Answer 2

除了查看代码之外，不要完全知道您要做什么。但这会给你最快的反应。至于在HashSet中找到值，它将是O（1）。

请尝试以下

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.HashSet;
import java.util.Set;

public class SalesTracking {
public static void main(String args[] ) throws Exception {

    BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

    int T = Integer.parseInt(br.readLine());
    if(T<=10 && T>=1)   {
    for (int i = 0; i < T; i++) {
       int count=0;
       int numOfPonds = Integer.parseInt(br.readLine());
        if(numOfPonds<=100000 && numOfPonds>=1){  
       String[] str ;
       //Map m = new HashMap();
       Set m = new HashSet();
        for(int j=0 ; j< numOfPonds ;j++)
                {   
                    str = br.readLine().split(" ");
                    int foodType = Integer.parseInt(str[0]);
                    int PokeType = Integer.parseInt(str[1]);
                    if(foodType<=1000000 && PokeType<=1000000 && foodType>=1 && PokeType>=1 && foodType != PokeType){
                        m.add(foodType);
                    if(!(m.contains(PokeType)))                        
                                    count++;
                    else if(m.contains(PokeType))
                        {   m.remove(PokeType);

                        }

                    }
                }
        }
       System.out.println(count);
      }
    }
    }
}

数据结构更快的contains（）操作？

2 个答案: